BZOJ 4012 HNOI2015 开店 动态树分治+二分
2015-04-29 12:44
344 查看
题目大意:给定一棵树,每个点有一个颜色,多次询问颜色在[l,r][l,r]区间内的所有点与某个点之间的距离之和,强制在线
没记错的话这题我知道的有三种解法来着?
(茴香豆的茴有四种写法泥萌知道嘛…?
1.线段树维护虚树
2.点分治+线段树
3.分块
第一种方法我不知道在线怎么搞= = (我并不知道怎么在虚树上进行点定位
第三种方法貌似内存过不去?
于是果断点分治+线段树
写完发现内存还是炸了= = O(nlog2n)O(nlog^2n)的内存说什么也过不去啊= =
后来发现既然维护的是和不是最值那还要线段树干嘛= =
直接开个vector维护前缀和直接二分不就好了= =
时间复杂度O(nlog2n)O(nlog^2n)
没记错的话这题我知道的有三种解法来着?
(茴香豆的茴有四种写法泥萌知道嘛…?
1.线段树维护虚树
2.点分治+线段树
3.分块
第一种方法我不知道在线怎么搞= = (我并不知道怎么在虚树上进行点定位
第三种方法貌似内存过不去?
于是果断点分治+线段树
写完发现内存还是炸了= = O(nlog2n)O(nlog^2n)的内存说什么也过不去啊= =
后来发现既然维护的是和不是最值那还要线段树干嘛= =
直接开个vector维护前缀和直接二分不就好了= =
时间复杂度O(nlog2n)O(nlog^2n)
[code]#include <vector> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define M 150100 #define B 400 using namespace std; typedef vector<pair<int,long long> > abcd; int n,m,A; int a[M]; long long last_ans; namespace Tree{ struct edge{ int to,f,next; bool ban; }table[M<<1]; int head[M],tot=1; int dpt[M],pos[M],T; int log_2[M<<1],min_dpt[M<<1][18]; void Add(int x,int y,int z) { table[++tot].to=y; table[tot].f=z; table[tot].next=head[x]; head[x]=tot; } void DFS(int x,int from) { int i; min_dpt[pos[x]=++T][0]=dpt[x]; for(i=head[x];i;i=table[i].next) if(table[i].to!=from) { dpt[table[i].to]=dpt[x]+table[i].f; DFS(table[i].to,x); min_dpt[++T][0]=dpt[x]; } } void Build_LCA() { int i,j; for(i=2;i<=T;i++) log_2[i]=log_2[i>>1]+1; for(j=1;j<=log_2[T];j++) for(i=1;i+(1<<j)-1<=T;i++) min_dpt[i][j]=min(min_dpt[i][j-1],min_dpt[i+(1<<j-1)][j-1]); } long long LCA_Depth(int x,int y) { x=pos[x];y=pos[y]; if(x>y) swap(x,y); int l=log_2[y-x+1]; return min(min_dpt[x][l],min_dpt[y-(1<<l)+1][l]); } long long Distance(int x,int y) { return dpt[x]+dpt[y]-2*LCA_Depth(x,y); } } pair<long long,int> operator + (const pair<long long,int> x,const pair<long long,int> y) { return make_pair(x.first+y.first,x.second+y.second); } pair<long long,int> operator - (const pair<long long,int> x,const pair<long long,int> y) { return make_pair(x.first-y.first,x.second-y.second); } /* struct Segtree{ Segtree *ls,*rs; long long sum1; int sum2; //sum1代表子树中所有点到该点的距离之和 //sum2代表子树中点的个数 void* operator new(size_t) { static Segtree *mempool,*C; if(C==mempool) mempool=(C=new Segtree[1<<15])+(1<<15); C->ls=C->rs=0x0; C->sum1=C->sum2=0; return C++; } friend void Insert(Segtree *&p,int x,int y,int pos,int val) { int mid=x+y>>1; if(!p) p=new Segtree; p->sum1+=val; p->sum2++; if(x==y) return ; if(pos<=mid) Insert(p->ls,x,mid,pos,val); else Insert(p->rs,mid+1,y,pos,val); } pair<long long,int> Query(Segtree *p,int x,int y,int l,int r) { int mid=x+y>>1; if(!p) return make_pair(0ll,0); if(x==l&&y==r) return make_pair(p->sum1,p->sum2); if(r<=mid) return Query(p->ls,x,mid,l,r); if(l>mid) return Query(p->rs,mid+1,y,l,r); return Query(p->ls,x,mid,l,mid) + Query(p->rs,mid+1,y,mid+1,r); } }; */ namespace Dynamic_TDC{ using namespace Tree; abcd sum1[M],sum2[M]; int fa[M]; int Get_Size(int x,int from) { int i,re=1; for(i=head[x];i;i=table[i].next) if(!table[i].ban&&table[i].to!=from) re+=Get_Size(table[i].to,x); return re; } int Get_Centre_Of_Gravity(int x,int from,int size,int &cg) { int i,re=1,flag=true; for(i=head[x];i;i=table[i].next) if(!table[i].ban&&table[i].to!=from) { int temp=Get_Centre_Of_Gravity(table[i].to,x,size,cg); if(temp<<1>size) flag=false; re+=temp; } if(size-re<<1>size) flag=false; if(flag) cg=x; return re; } void DFS(int x,int from,int dpt,abcd &sum1,abcd &sum2) { int i; sum1.push_back(pair<int,long long>(a[x],dpt)); sum2.push_back(pair<int,long long>(a[x],dpt)); for(i=head[x];i;i=table[i].next) if(!table[i].ban&&table[i].to!=from) DFS(table[i].to,x,dpt+table[i].f,sum1,sum2); } int Tree_Divide_And_Conquer(int x) { abcd::iterator it; int i,j,size=Get_Size(x,0); Get_Centre_Of_Gravity(x,0,size,x); sum1[x].push_back(pair<int,long long>(a[x],0)); for(i=head[x];i;i=table[i].next) if(!table[i].ban) { table[i].ban=table[i^1].ban=true; abcd temp; DFS(table[i].to,0,table[i].f,temp,sum1[x]); int y=Tree_Divide_And_Conquer(table[i].to); sum2[y]=temp;fa[y]=x; sum2[y].push_back(pair<int,long long>(-1,0)); sort(sum2[y].begin(),sum2[y].end()); for(j=1;j<sum2[y].size();j++) sum2[y][j].second+=sum2[y][j-1].second; } sum1[x].push_back(pair<int,long long>(-1,0)); sort(sum1[x].begin(),sum1[x].end()); for(j=1;j<sum1[x].size();j++) sum1[x][j].second+=sum1[x][j-1].second; return x; } pair<long long,int> Query(abcd &a,int l,int r) { if(a.empty()) return pair<long long,int>(0,0); abcd::iterator it1=lower_bound(a.begin(),a.end(),pair<int,long long>(l,0)); abcd::iterator it2=lower_bound(a.begin(),a.end(),pair<int,long long>(r+1,0)); it1--;it2--; return make_pair(it2->second-it1->second,it2-it1); } long long Query(int x,int l,int r) { int i; long long re=Query(sum1[x],l,r).first; for(i=x;fa[i];i=fa[i]) { pair<long long,int> temp=Query(sum1[fa[i]],l,r)-Query(sum2[i],l,r); re+=temp.first+temp.second*Distance(x,fa[i]); } return re; } } int main() { using namespace Tree; using namespace Dynamic_TDC; int i,x,y,z; cin>>n>>m>>A; for(i=1;i<=n;i++) scanf("%d",&a[i]); for(i=1;i<n;i++) { scanf("%d%d%d",&x,&y,&z); Add(x,y,z);Add(y,x,z); } DFS(1,0);Build_LCA(); Tree_Divide_And_Conquer(1); for(i=1;i<=m;i++) { scanf("%d%d%d",&x,&y,&z); y=(y+last_ans)%A; z=(z+last_ans)%A; if(y>z) swap(y,z); printf("%lld\n",last_ans=Query(x,y,z)); } return 0; }
相关文章推荐
- 【BZOJ4012】[HNOI2015]开店 动态树分治+二分
- bzoj4012 [HNOI2015]开店(动态点分治+二分+STL/树链剖分+主席树)
- BZOJ 4012 【HNOI2015】 开店
- [BZOJ4012][HNOI2015]开店(动态点分治)
- bzoj 4012: [HNOI2015]开店 主席树
- bzoj4012 [HNOI2015]开店
- 【BZOJ 4012】[HNOI2015]开店
- bzoj 4012: [HNOI2015]开店 (树链剖分+主席树)
- bzoj 4012: [HNOI2015]开店
- [动态树分治] BZOJ4012 [HNOI2015]开店
- BZOJ 4012 HNOI 2015 开店(shop) 一道简单的点剖题
- 【BZOJ】4012: [HNOI2015]开店
- BZOJ 4012: [HNOI2015]开店
- [BZOJ]4012: [HNOI2015]开店 树链剖分+主席树(线段树合并)
- BZOJ 4012: [HNOI2015]开店 -- 动态树分治
- BZOJ4012: [HNOI2015]开店 重链剖分 可持久化线段树
- 【bzoj4012】[HNOI2015]开店 动态点分治+STL-vector
- BZOJ4012 [HNOI2015]开店
- bzoj4012 [HNOI2015]开店
- bzoj4012[HNOI2015]开店