#include <bits/stdc++.h> #define PB push_back #define ST first #define ND second #define _ ios_base::sync_with_stdio(0); cin.tie(0); //mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count()); using namespace std; using ll = long long; using pi = pair<int,int>; using vi = vector<int>; const int nax = 300 * 1000 + 10, mod = 1e9 + 7; int n, q; vi V[nax]; int A[nax]; int dep[nax], p[nax], ex[nax], d[nax]; bool visited[nax]; int last = 1; void create() { int x = last; ex[dep[x]] = x; if(dep[x] == n) return; for(int i = 1; i <= A[dep[x]]; ++i) { dep[++last] = dep[x] + 1; p[last] = x; V[last].PB(x); V[x].PB(last); create(); } } int D, niemoge; int find_sec(int x) { if(dep[x] == D) return x; for(int nbh : V[x]) if(nbh != p[x] && nbh != niemoge) { return find_sec(nbh); } } int all; void dfs(int x, int dis) { visited[x] = 1; d[x] += dis; all = (all + dis) % mod; for(int nbh : V[x]) if(!visited[nbh]) { dfs(nbh, dis+1); } } int main() {_ cin >> n >> q; for(int i = 1; i < n; ++i) { cin >> A[i]; } dep[1] = 1; create(); while(q--) { int a, b, c; cin >> a >> b >> c; int x = ex[a]; int y = x, block = -1; while(dep[y] != c) { block = y; y = p[y]; } D = b; niemoge = block; y = find_sec(y); for(int i = 1; i <= last; ++i) { visited[i] = 0; d[i] = 0; } dfs(x, 0); for(int i = 1; i <= last; ++i) { visited[i] = 0; } all = 0; dfs(y, 0); sort(d + 1, d + last + 1); int ans = 0; for(int i = last; i >= 1; --i) { if((last - i) %2 ==0) { ans = (ans + d[i]) % mod; } } ans = (ans - all) % mod; if(ans < 0) ans += mod; cout << ans << "\n"; } }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | #include <bits/stdc++.h> #define PB push_back #define ST first #define ND second #define _ ios_base::sync_with_stdio(0); cin.tie(0); //mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count()); using namespace std; using ll = long long; using pi = pair<int,int>; using vi = vector<int>; const int nax = 300 * 1000 + 10, mod = 1e9 + 7; int n, q; vi V[nax]; int A[nax]; int dep[nax], p[nax], ex[nax], d[nax]; bool visited[nax]; int last = 1; void create() { int x = last; ex[dep[x]] = x; if(dep[x] == n) return; for(int i = 1; i <= A[dep[x]]; ++i) { dep[++last] = dep[x] + 1; p[last] = x; V[last].PB(x); V[x].PB(last); create(); } } int D, niemoge; int find_sec(int x) { if(dep[x] == D) return x; for(int nbh : V[x]) if(nbh != p[x] && nbh != niemoge) { return find_sec(nbh); } } int all; void dfs(int x, int dis) { visited[x] = 1; d[x] += dis; all = (all + dis) % mod; for(int nbh : V[x]) if(!visited[nbh]) { dfs(nbh, dis+1); } } int main() {_ cin >> n >> q; for(int i = 1; i < n; ++i) { cin >> A[i]; } dep[1] = 1; create(); while(q--) { int a, b, c; cin >> a >> b >> c; int x = ex[a]; int y = x, block = -1; while(dep[y] != c) { block = y; y = p[y]; } D = b; niemoge = block; y = find_sec(y); for(int i = 1; i <= last; ++i) { visited[i] = 0; d[i] = 0; } dfs(x, 0); for(int i = 1; i <= last; ++i) { visited[i] = 0; } all = 0; dfs(y, 0); sort(d + 1, d + last + 1); int ans = 0; for(int i = last; i >= 1; --i) { if((last - i) %2 ==0) { ans = (ans + d[i]) % mod; } } ans = (ans - all) % mod; if(ans < 0) ans += mod; cout << ans << "\n"; } } |