首页 > 代码库 > hdu 5044 树区间操作最后输出/ lca+dfs
hdu 5044 树区间操作最后输出/ lca+dfs
题意:一棵树,俩种操作:1 有路径上的所有点加vi,2所有边加vi。
先离线求出所有询问的lca,再遍历询问一次,点+vi,lca-2*vi ,最后dfs从叶子扫上来一次,最后再祖先点补上即可。用了输入挂。复杂度(n+m).
#pragma comment(linker,"/STACK:10240000000000,10240000000000") #include<iostream> #include<cstdio> #include<cstdlib> using namespace std; const int maxv=100005,maxe=200015; int head[maxv];int nume=0;long long e[maxe][3]; void inline adde(int i,int j) //边 { e[nume][0]=j;e[nume][1]=head[i];head[i]=nume; e[nume++][2]=0; e[nume][0]=i;e[nume][1]=head[j];head[j]=nume; e[nume++][2]=0; } int headq[maxv];int numq=0;long long q[maxe][5]; void inline addq(int x,int y,long long vals,int flag) //询问 { q[numq][0]=y;q[numq][1]=headq[x];headq[x]=numq; q[numq][2]=-1;q[numq][3]=vals;q[numq++][4]=flag; q[numq][0]=x;q[numq][1]=headq[y];headq[y]=numq; q[numq][2]=-1;q[numq][3]=vals;q[numq++][4]=flag; } int fa[maxv]; int find(int x) { if(x!=fa[x]) return fa[x]=find(fa[x]); return fa[x]; } int ve[maxv]; int n,m; int vis[maxv];int vis2[maxv]; long long val[maxv]; int ise[maxe]; void tarjan(int u,int father) //离线求lca { for(int j=head[u];j!=-1;j=e[j][1]) { int v=e[j][0]; if(!vis[v]&&v!=father) { ve[v]=j; ise[j]=1; tarjan(v,u); fa[v]=u; } } vis[u]=1; for(int j=headq[u];j!=-1;j=q[j][1]) { if(vis[q[j][0]]) { q[j^1][2]=q[j][2]=find(q[j][0]); } } } void dfs(int u) //求出所有点边情况 { for(int j=head[u];j!=-1;j=e[j][1]) { int v=e[j][0]; if(!vis2[v]) { vis2[v]=1; dfs(v); val[u]+=val[v]; e[ve[u]][2]+=e[ve[v]][2]; } } } int vis_e[maxe]; long long anster[maxv]; void init() { nume=numq=0; for(int i=0;i<maxv;i++) { head[i]=headq[i]=-1; fa[i]=i; val[i]=vis2[i]=vis[i]=0; ve[i]=2*n; anster[i]=0; } for(int i=0;i<maxe;i++) { ise[i]=0; vis_e[i]=0; } } template <class T> //输入挂 inline bool scan(T &ret) { char c; int sgn; if(c=getchar(),c==EOF) return 0; //EOF while(c!='-'&&(c<'0'||c>'9')) c=getchar(); sgn=(c=='-')?-1:1; ret=(c=='-')?0:(c-'0'); while(c=getchar(),c>='0'&&c<='9') ret=ret*10+(c-'0'); ret*=sgn; return 1; } /*inline void prints(long long x) { if(x>9) prints(x/10); putchar(x%10+'0'); }*/ int main() { int T;int cnt=1; scan(T); //scanf("%d",&T); while(T--) { //scanf("%d%d",&n,&m); scan(n);scan(m); init(); int aa,bb,cc; for(int i=0;i<n-1;i++) { // scanf("%d%d",&aa,&bb); scan(aa);scan(bb); adde(aa,bb); } int tx; for(int i=0;i<m;i++) { getchar();getchar();getchar();//getchar(); //scanf("ADD"); scan(tx); scan(aa);scan(bb);scan(cc); // scanf("%d%d%d%d",&tx,&aa,&bb,&cc); if(tx==1) addq(aa,bb,cc,1); else addq(aa,bb,cc,0); } tarjan(n/2+1,-1); for(int i=1;i<=n;i++) for(int j=headq[i];j!=-1;j=q[j][1]) { if(!vis_e[j]) { vis_e[j]=vis_e[j^1]=1; if(q[j][4]) { val[i]+=q[j][3]; val[q[j][0]]+=q[j][3]; val[q[j][2]]-=q[j][3]*2; anster[q[j][2]]+=q[j][3]; } else { e[ve[i]][2]+=q[j][3]; e[ve[q[j][0]]][2]+=q[j][3]; e[ve[q[j][2]]][2]-=2*q[j][3]; } } } vis2[n/2+1]=1; dfs(n/2+1); printf("Case #%d:\n",cnt++); for(int i=1;i<n;i++) printf("%I64d ",val[i]+anster[i]); printf("%I64d\n",val[n]+anster[n]); for(int j=0;j<nume;j++) if(ise[j]) { if(j==nume-2||j==nume-1)printf("%I64d",e[j][2]); else printf("%I64d ",e[j][2]); } puts(""); } return 0; }
hdu 5044 树区间操作最后输出/ lca+dfs
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。