题目大意:
给出一个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的最近公共祖先的深度之和)
这题看见了直接卡壳。。。然后看了题解才搞懂要怎么写。。。
直接复制gconeice的题解吧
显然,暴力求解的复杂度是无法承受的。
考虑这样的一种暴力,我们把 z 到根上的点全部打标记,对于 l 到 r 之间的点,向上搜索到第一个有标记的点求出它的深度统计答案。观察到,深度其实就是上面有几个已标记了的点(包括自身)。所以,我们不妨把 z 到根的路径上的点全部 +1,对于 l 到 r 之间的点询问他们到根路径上的点权和。仔细观察上面的暴力不难发现,实际上这个操作具有叠加性,且可逆。也就是说我们可以对于 l 到 r 之间的点 i,将 i 到根的路径上的点全部 +1, 转而询问 z 到根的路径上的点(包括自身)的权值和就是这个询问的答案。把询问差分下,也就是用 [1, r] − [1, l − 1] 来计算答案,那么现在我们就有一个明显的解法。从 0 到 n − 1 依次插入点 i,即将 i 到根的路径上的点全部+1。离线询问答案即可。我们现在需要一个数据结构来维护路径加和路径求和,显然树链剖分或LCT 均可以完成这个任务。树链剖分的复杂度为 O((n + q)· log n · log n),LCT的复杂度为 O((n + q)· log n),均可以完成任务。至此,题目已经被我们完美解决。
写的很详细,我就不累述了
同样是14年的省选,同样是树链剖分,为啥辽宁省的就如此牛B,吉林省的就是水题。。。差距好大0.0
取模的话数字不是很大 可以开long long 最后取一下模就好了 中途不要取模 弄成负的WA了我一下午0.0 又一下午0.0 又一下午0.0 啊啊啊啊啊啊啊啊啊
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define ls tree[p].lson #define rs tree[p].rson #define root 0,1,n #define M 50100 using namespace std; typedef long long ll; struct abcd{ int to,next; }table[M]; int head[M],tot; struct Query{ int num,z,pos; ll ans; }q[M<<1]; struct Tree{ ll num,mark; int lson,rson; }tree[M<<1];int treetot; int n,m,fa[M],son[M],dpt[M],siz[M],pos[M],top[M],cnt; bool operator < (Query x,Query y) { return x.num < y.num ; } int cmp(Query x,Query y) { if(x.pos==y.pos) return x.num < y.num ; return x.pos < y.pos ; } void add(int x,int y) { table[++tot].to=y; table[tot].next=head[x]; head[x]=tot; } void Build_tree(int p,int x,int y) { int mid=x+y>>1; if(x==y) return ; ls=++treetot; rs=++treetot; Build_tree(ls,x,mid); Build_tree(rs,mid+1,y); } void change(int p,int x,int y,int l,int r) { int mid=x+y>>1; if(x==l&&y==r) { tree[p].num+=(r-l+1); tree[p].mark++; return ; } if(tree[p].mark) { tree[ls].num+=(mid-x+1)*tree[p].mark; tree[rs].num+=(y-mid)*tree[p].mark; tree[ls].mark+=tree[p].mark; tree[rs].mark+=tree[p].mark; tree[p].mark=0; } if(r<=mid) change(ls,x,mid,l,r); else if(l>mid) change(rs,mid+1,y,l,r); else change(ls,x,mid,l,mid),change(rs,mid+1,y,mid+1,r); tree[p].num=tree[ls].num+tree[rs].num; } ll getans(int p,int x,int y,int l,int r) { int mid=x+y>>1; if(x==l&&y==r) return tree[p].num; if(tree[p].mark) { tree[ls].num+=(mid-x+1)*tree[p].mark; tree[rs].num+=(y-mid)*tree[p].mark; tree[ls].mark+=tree[p].mark; tree[rs].mark+=tree[p].mark; tree[p].mark=0; } if(r<=mid) return getans(ls,x,mid,l,r); if(l>mid) return getans(rs,mid+1,y,l,r); return getans(ls,x,mid,l,mid)+getans(rs,mid+1,y,mid+1,r); } void dfs1(int x) { int i; dpt[x]=dpt[fa[x]]+1; siz[x]=1; for(i=head[x];i;i=table[i].next) { fa[table[i].to]=x; dfs1(table[i].to); if(siz[table[i].to]>siz[son[x]]) son[x]=table[i].to; siz[x]+=siz[table[i].to]; } } void dfs2(int x) { int i; if(son[fa[x]]==x) top[x]=top[fa[x]]; else { top[x]=x; for(i=x;i;i=son[i]) pos[i]=++cnt; } for(i=head[x];i;i=table[i].next) dfs2(table[i].to); } void update(int x) { int fx=top[x]; while(x) { change(root,pos[fx],pos[x]); x=fa[fx]; fx=top[x]; } } ll query(int x) { ll re=0; int fx=top[x]; while(x) { re+=getans(root,pos[fx],pos[x]); x=fa[fx]; fx=top[x]; } return re; } int main() { int i,j,x,y,z; cin>>n>>m; for(i=2;i<=n;i++) scanf("%d",&x),add(x+1,i); dfs1(1); dfs2(1); for(i=1;i<=m;i++) { scanf("%d%d%d",&x,&y,&z); x++;y++;z++; q[i+i-1].num=x-1; q[i+i-1].z=z; q[i+i-1].pos=i; q[i<<1].num=y; q[i<<1].z=z; q[i<<1].pos=i; } sort(q+1,q+m+m+1); Build_tree(root); j=1; for(i=0;i<=n;i++) { update(i); for(;q[j].num==i;j++) q[j].ans=query(q[j].z); } sort(q+1,q+m+m+1,cmp); for(i=1;i<=m;i++) printf("%lld\n", ( q[i<<1].ans - q[i+i-1].ans ) % 201314 ); }