#pragma GCC optimize("Ofast")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
using namespace std;
#define mp make_pair
#define eb emplace_back
#define pb push_back
#define e1 first
#define e2 second
#define uint unsigned int
#define ll long long
#define ull unsigned long long
#define ld long double
#define float long double
#define size(x) (int)x.size()
#define satori int testCases; cin>>testCases; while(testCases--)
#define fastio ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define all(r) begin(r),end(r)
#define time chrono::high_resolution_clock().now().time_since_epoch().count()
typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> ordered_set;
mt19937 rng(chrono::high_resolution_clock().now().time_since_epoch().count());
///////////////////
#define debug if(1)
///////////////////
const int MAXN=3e5+10;
const int mod=1e9+7;
const ll inv=5e8+4;
int nodes=0;
vector<int> g[MAXN];
int a[MAXN];
int cost[MAXN];
int cnt[MAXN*2];
int my_pos[MAXN];
void prep(int id,int pos,int max_pos){
my_pos[id]=pos;
if(pos==max_pos)
return;
for(int i=0;i<a[pos];i++){
nodes++;
g[nodes].eb(id);
g[id].eb(nodes);
prep(nodes,pos+1,max_pos);
}
}
int get(int u,int req_pos,bool lewy){
if(my_pos[u]==req_pos)
return u;
if(lewy)
return get(g[u][1],req_pos,lewy);
return get(g[u][size(g[u])-1],req_pos,lewy);
}
int ans;
void dfs(int u,int d,int col,int p=-1){
if(!col){
ans+=d;
if(ans>=mod)
ans-=mod;
}
else{
ans-=d;
if(ans<0)
ans+=mod;
}
cost[u]+=d;
for(auto v:g[u])
if(v!=p&&v!=0)
dfs(v,d+1,col,u);
}
int32_t main()
{
fastio;
int n,q;
cin>>n>>q;
for(int i=1;i<n;i++)
cin>>a[i];
g[1].eb(0);
nodes=1;
prep(1,1,n);
int l,r,lca,turn;
ll ansans;
while(q--){
cin>>l>>r>>lca;
lca=get(1,lca,0);
l=get(lca,l,0);
r=get(lca,r,1);
fill(cost+1,cost+nodes+1,0);
fill(cnt,cnt+nodes+nodes+1,0);
ans=0;
dfs(l,0,0);
dfs(r,0,1);
for(int i=1;i<=nodes;i++)
cnt[cost[i]]++;
turn=0;
for(int i=nodes*2;i>=0;i--){
while(cnt[i]>0){
if(!turn){
ans+=i;
if(ans>=mod)
ans-=mod;
}
else{
ans-=i;
if(ans<0)
ans+=mod;
}
cnt[i]--;
turn^=1;
}
}
ansans=ans;
ansans=(ansans*inv)%mod;
cout<<ansans<<'\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 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 | #pragma GCC optimize("Ofast") #pragma GCC optimization ("O3") #pragma GCC optimization ("unroll-loops") #include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> using namespace __gnu_pbds; using namespace std; #define mp make_pair #define eb emplace_back #define pb push_back #define e1 first #define e2 second #define uint unsigned int #define ll long long #define ull unsigned long long #define ld long double #define float long double #define size(x) (int)x.size() #define satori int testCases; cin>>testCases; while(testCases--) #define fastio ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0) #define all(r) begin(r),end(r) #define time chrono::high_resolution_clock().now().time_since_epoch().count() typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> ordered_set; mt19937 rng(chrono::high_resolution_clock().now().time_since_epoch().count()); /////////////////// #define debug if(1) /////////////////// const int MAXN=3e5+10; const int mod=1e9+7; const ll inv=5e8+4; int nodes=0; vector<int> g[MAXN]; int a[MAXN]; int cost[MAXN]; int cnt[MAXN*2]; int my_pos[MAXN]; void prep(int id,int pos,int max_pos){ my_pos[id]=pos; if(pos==max_pos) return; for(int i=0;i<a[pos];i++){ nodes++; g[nodes].eb(id); g[id].eb(nodes); prep(nodes,pos+1,max_pos); } } int get(int u,int req_pos,bool lewy){ if(my_pos[u]==req_pos) return u; if(lewy) return get(g[u][1],req_pos,lewy); return get(g[u][size(g[u])-1],req_pos,lewy); } int ans; void dfs(int u,int d,int col,int p=-1){ if(!col){ ans+=d; if(ans>=mod) ans-=mod; } else{ ans-=d; if(ans<0) ans+=mod; } cost[u]+=d; for(auto v:g[u]) if(v!=p&&v!=0) dfs(v,d+1,col,u); } int32_t main() { fastio; int n,q; cin>>n>>q; for(int i=1;i<n;i++) cin>>a[i]; g[1].eb(0); nodes=1; prep(1,1,n); int l,r,lca,turn; ll ansans; while(q--){ cin>>l>>r>>lca; lca=get(1,lca,0); l=get(lca,l,0); r=get(lca,r,1); fill(cost+1,cost+nodes+1,0); fill(cnt,cnt+nodes+nodes+1,0); ans=0; dfs(l,0,0); dfs(r,0,1); for(int i=1;i<=nodes;i++) cnt[cost[i]]++; turn=0; for(int i=nodes*2;i>=0;i--){ while(cnt[i]>0){ if(!turn){ ans+=i; if(ans>=mod) ans-=mod; } else{ ans-=i; if(ans<0) ans+=mod; } cnt[i]--; turn^=1; } } ansans=ans; ansans=(ansans*inv)%mod; cout<<ansans<<'\n'; } } |
English