首页 > 代码库 > BZOJ 3631 JLOI2014 松鼠的新家 树链剖分/LCA
BZOJ 3631 JLOI2014 松鼠的新家 树链剖分/LCA
题目大意:给定一棵无根树和一个序列,在这个序列上依次遍历,求每个点的访问次数(最后一个点的访问次数要-1)
树链剖分的裸题……考场上我还是一个弱渣,啥也不会,暴力得了50分,剩下两道题爆零了。。。而且30W深搜爆栈,人生第一次手写了系统栈。。
回来因为这题的原因去学了树链剖分 结果没学明白 每条重链单独开了一棵线段树 常数大的要死
高一时写的代码。。。还是别看了,拿去对拍可以,阅读性欠佳
#include<stdio.h> #include<stdlib.h> #include<string.h> #define M 300100 #define min(x,y) x<y?x:y typedef struct abcd{int xx,mark,l,r;abcd*ls,*rs;} abcd;abcd*tree[M]; typedef struct abcde{int to;abcde*next;} abcde;abcde*head[M]; void add(int x,int y){abcde*temp=(abcde*)malloc(sizeof(abcde));temp->to=y;temp->next=head[x];head[x]=temp;} void swap(int &x,int &y){int z=x;x=y;y=z;} int fa[M],siz[M],dpt[M],son[M],top[M],a[M],ans; void Build_tree(abcd*p,int l,int r) { int mid=l+r>>1; p->l=l;p->r=r;p->xx=p->mark=0;p->ls=p->rs=NULL; if(l==r)return ; p->ls=(abcd*)malloc(sizeof(abcd)); p->rs=(abcd*)malloc(sizeof(abcd)); Build_tree(p->ls,l,mid); Build_tree(p->rs,mid+1,r); } void Up_load(abcd*p,int x,int y) { int mid=p->l+p->r>>1; if(p->l==x&&p->r==y){p->xx+=p->r-p->l+1;p->mark++;return ;} if(y<=mid)Up_load(p->ls,x,y); else if(x>mid)Up_load(p->rs,x,y); else Up_load(p->ls,x,mid),Up_load(p->rs,mid+1,y); p->xx+=y-x+1; } int search(abcd*p,int x) { int mid=p->l+p->r>>1; if(p->l==x&&p->r==x)return p->xx; p->ls->xx+=p->mark*(p->ls->r-p->ls->l+1); p->rs->xx+=p->mark*(p->rs->r-p->rs->l+1); p->ls->mark+=p->mark; p->rs->mark+=p->mark; p->mark=0; if(x<=mid)return search(p->ls,x);return search(p->rs,x); } int q[M],r,h; int maxsiz[M]; void bfs() { abcde*temp; int x; q[++h]=1; while(r!=h) { r++; x=q[r]; dpt[x]=dpt[fa[x]]+1; siz[x]=1; for(temp=head[x];temp;temp=temp->next) { if(temp->to==fa[x])continue; fa[temp->to]=x; q[++h]=temp->to; } } for(;r>=2;r--) { x=q[r]; siz[fa[x]]+=siz[x]; if(siz[x]>maxsiz[fa[x]])maxsiz[fa[x]]=siz[x],son[fa[x]]=x; } for(r=1;r<=h;r++) { x=q[r]; if (son[fa[x]]!=x)top[x]=x,tree[x]=(abcd*)malloc(sizeof(abcd)); else top[x]=top[fa[x]]; if(son[x]==0)Build_tree(tree[top[x]],1,dpt[x]-dpt[top[x]]+1); } } int n; int main() { int i,x,y,f1,f2; scanf("%d",&n); for(i=1;i<=n;i++)scanf("%d",&a[i]); for(i=1;i<n;i++)scanf("%d%d",&x,&y),add(x,y),add(y,x); bfs(); for(i=1;i<n;i++) { x=a[i];y=a[i+1]; f1=top[x];f2=top[y]; while(f1!=f2) { if(dpt[f1]<dpt[f2])swap(x,y),swap(f1,f2); Up_load(tree[f1],1,dpt[x]-dpt[f1]+1); x=fa[f1]; f1=top[x]; } if(dpt[x]<dpt[y])swap(x,y); Up_load(tree[f1],dpt[y]-dpt[f1]+1,dpt[x]-dpt[f1]+1); } for(i=1;i<=n;i++) printf("%d\n",search(tree[top[i]],dpt[i]-dpt[top[i]]+1)-1+(i==a[1])); }
后来经过讨论其实这题LCA就能过。。。每次移动时在起始点和终点各打一个start标记,在LCA和LCA的父节点处各上打一个end标记,然后深搜,start标记一直上传,遇到end标记就停止,最后再处理一下就行
我写的是TarjanLCA 这样询问深搜一遍处理 时间复杂度O(n)
#include<stdio.h> #include<stdlib.h> #include<string.h> #define M 300100 struct abcd{ int to,next; }table[M<<2]; int head[M],query[M],tot; void add(int x,int y) { table[++tot].to=y; table[tot].next=head[x]; head[x]=tot; } void qadd(int x,int y) { table[++tot].to=y; table[tot].next=query[x]; query[x]=tot; } int fa[M],f[M],a[M],v[M]; int start[M],end[M]; int n; int find(int x) { if(!f[x]||f[x]==x) return f[x]=x; return f[x]=find(f[x]); } void Tarjan(int x) { int i; for(i=head[x];i;i=table[i].next) if(table[i].to!=fa[x]) fa[table[i].to]=x,Tarjan(table[i].to),f[table[i].to]=x; v[x]=1; for(i=query[x];i;i=table[i].next) if(v[table[i].to]) end[ fa[ find(table[i].to) ] ]++,end[ find(table[i].to) ]++; start[x]-=end[x]; start[fa[x]]+=start[x]; } int main() { int i,x,y; scanf("%d",&n); for(i=1;i<=n;i++) { scanf("%d",&a[i]); if(i^1) { qadd(a[i],a[i-1]); qadd(a[i-1],a[i]); start[ a[i] ]++; start[ a[i-1] ]++; } } for(i=1;i<n;i++) scanf("%d%d",&x,&y),add(x,y),add(y,x); Tarjan(1); for(i=1;i<=n;i++) printf("%d\n", start[i] - (i!=a[1]) ); }
BZOJ 3631 JLOI2014 松鼠的新家 树链剖分/LCA
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。