bzoj3626

链剖。。。题解看hz的。。。。离线搞

  1 #include<bits/stdc++.h>
  2 #define lowbit(a) ((a)&(-(a)))
  3 #define l(a) ((a)<<1)
  4 #define r(a) ((a)<<1|1)
  5 #define clr(a,x) memset(a,x,sizeof(a))
  6 #define rep(i,l,r) for(int i=l;i<(r);i++)
  7 #define Rep(i,a) rep(i,0,e[a].size())
  8 typedef long long ll;
  9 using namespace std;
 10 int read()
 11 {
 12     char c=getchar();
 13     int ans=0,f=1;
 14     while(!isdigit(c)){
 15         if(c=='-') f=-1;
 16         c=getchar();
 17     }
 18     while(isdigit(c)){
 19         ans=ans*10+c-'0';
 20         c=getchar();
 21     }
 22     return ans*f;
 23 }
 24 struct node{
 25     int l,r,sum,add;
 26 };
 27 struct query{
 28     int x,p;
 29     bool f;
 30     inline bool operator<(const query&A)const{
 31         return x<A.x;
 32     }
 33 };
 34 const int maxn=50009,mod=201314;
 35 int ans[maxn],n,m,dfstime=0,Top,g[maxn],size[maxn],f[maxn],son[maxn],top[maxn],dep[maxn],id[maxn];
 36 query q[maxn<<1];
 37 node x[maxn<<2];
 38 vector<int>e[maxn];
 39 void dfs(int k){
 40     size[k]=1;
 41     Rep(i,k){
 42         int to=e[k][i];
 43         if(to==f[k]) continue;
 44         f[to]=k;
 45         dep[to]=dep[k]+1;
 46         dfs(to);
 47         size[k]+=size[to];
 48         if(!son[k]||size[son[k]]<size[to]) son[k]=to;
 49     }
 50 }
 51 void Dfs(int k){
 52     top[k]=Top;
 53     id[k]=++dfstime;
 54     if(son[k]) Dfs(son[k]);
 55     Rep(i,k){
 56         int to=e[k][i];
 57         if(!id[to]) Dfs(Top=to);
 58     }
 59 }
 60 void maintain(int k){
 61     x[k].sum=x[l(k)].sum+x[r(k)].sum;
 62 }
 63 void pushdown(int k){
 64     if(x[k].l!=x[k].r&&x[k].add){
 65         x[l(k)].sum+=(x[l(k)].r-x[l(k)].l+1)*x[k].add;
 66         x[r(k)].sum+=(x[r(k)].r-x[r(k)].l+1)*x[k].add;
 67         x[l(k)].add+=x[k].add;
 68         x[r(k)].add+=x[k].add;
 69     }
 70     x[k].add=0;
 71 }
 72 void build(int k,int l,int r){
 73     x[k].l=l,x[k].r=r,x[k].sum=x[k].add=0;
 74     if(l==r) return;
 75     int mid=(l+r)>>1;
 76     build(l(k),l,mid);
 77     build(r(k),mid+1,r);
 78     maintain(k);
 79 }
 80 void modify(int k,int l,int r,int t){
 81     pushdown(k);
 82     if(x[k].l==l&&x[k].r==r){
 83         x[k].sum+=(r-l+1)*t;
 84         x[k].add+=t;
 85         return;
 86     }
 87     int mid=(x[k].l+x[k].r)>>1;
 88     if(r<=mid) modify(l(k),l,r,t);
 89     else if(l>mid) modify(r(k),l,r,t);
 90     else{
 91         modify(l(k),l,mid,t);
 92         modify(r(k),mid+1,r,t);
 93     }
 94     maintain(k);
 95 }
 96 int find(int k,int l,int r){
 97     pushdown(k);
 98     if(x[k].l==l&&x[k].r==r) return x[k].sum;
 99     int mid=(x[k].l+x[k].r)>>1;
100     if(r<=mid) return find(l(k),l,r);
101     if(l>mid) return find(r(k),l,r);
102     return (find(l(k),l,mid)+find(r(k),mid+1,r));
103 }
104 void add(int u,int v,int t){
105     while(top[u]!=top[v]){
106         if(dep[top[u]]<dep[top[v]]) swap(u,v);
107         modify(1,id[top[u]],id[u],t);
108         u=f[top[u]];
109     }
110     if(dep[u]>dep[v]) swap(u,v);
111     modify(1,id[u],id[v],t);
112 }
113 int sum(int u,int v){
114     int ans=0;
115     while(top[u]!=top[v]){
116         if(dep[top[u]]<dep[top[v]]) swap(u,v);
117         ans+=find(1,id[top[u]],id[u]);
118         u=f[top[u]];
119     }
120     if(dep[u]>dep[v]) swap(u,v);
121     ans+=find(1,id[u],id[v]);
122     return ans%mod;
123 }
124 void init(){
125     dep[0]=1;
126     dfs(0);
127     Dfs(Top=0);
128     build(1,1,n);
129 }
130 int main()
131 {    
132     n=read(),m=read();
133     rep(i,1,n){
134         int to=read();
135         e[to].push_back(i);
136         e[i].push_back(to);
137     }
138     init();
139     int cnt=0;
140     rep(i,0,m){
141         int l=read(),r=read();g[i]=read();
142         q[cnt].x=l-1;q[cnt].p=i;q[cnt++].f=0;
143         q[cnt].x=r;q[cnt].p=i;q[cnt++].f=1;
144     }
145     m<<=1;
146     sort(q,q+m);
147     cnt=-1;
148     rep(i,0,m){
149         while(cnt<q[i].x){    
150             cnt++;
151             add(cnt,0,1);
152         }
153         int t=sum(g[q[i].p],0);
154         ans[q[i].p]=ans[q[i].p]+t*(q[i].f?1:-1);
155     }
156     rep(i,0,m>>1) printf("%d
",(ans[i]+mod)%mod);
157     return 0;
158 }

View Code

3626: [LNOI2014]LCA

Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 947 Solved: 325
[Submit][Status][Discuss]

Description

给出一个n个节点的有根树（编号为0到n-1，根节点为0）。一个点的深度定义为这个节点到根的距离+1。
设dep[i]表示点i的深度，LCA(i,j)表示i与j的最近公共祖先。
有q次询问，每次询问给出l r z，求sigma_{l<=i<=r}dep[LCA(i,z)]。
（即，求在[l,r]区间内的每个节点i与z的最近公共祖先的深度之和）

Input

第一行2个整数n q。
接下来n-1行，分别表示点1到点n-1的父节点编号。
接下来q行，每行3个整数l r z。

Output

输出q行，每行表示一个询问的答案。每个答案对201314取模输出

Sample Input

5 2
0
0
1
1
1 4 3
1 4 2

Sample Output

8
5

HINT

共5组数据，n与q的规模分别为10000,20000,30000,40000,50000。

Source

数据已加强 by saffah

[Submit][Status][Discuss]