uoj261 天天爱跑步
Posted MashiroSky
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了uoj261 天天爱跑步相关的知识,希望对你有一定的参考价值。
http://uoj.ac/problem/261 (题目链接)
题意
给出一棵树,给出一些起点和终点,没走一条路径耗费时间1,每个节点上有一个权值w,问有多少条路径经过这个节点时所用的时间恰好是w。
Solution
转自:http://blog.csdn.net/haarmony/article/details/53259338
约定第${i}$个人起终点的${lca(s[i],t[i])}$为${lca[i]}$,点${i}$深度为${deep[i]}$
- 考虑可能对点${u}$有贡献的第${i}$个跑步者,${lca[i]}$肯定在${u}$或者${u}$上方,否则不经过${u}$点,即${lca[i]==u || lca[i]∉subtree[u]}$
- 基于此前提,只有两种情况(有重合部分):${s[i]∈subtree[u]}$,${t[i]∈subtree[u]}$
分类讨论能被${u}$看见的点${v}$的情况
1.${v}$为${i}$人起点${s[i]}$:${deep[s[i]]-w[u]==dep[u]}$
2.${v}$为${i}$人终点${t[i]}$:${deep[s[i]]+deep[t[i]]-2*deep[lca[i]]-(deep[t[i]]-deep[u])==w[u]}$,即${deep[s[i]]-2*deep[lca[i]]==w[u]-deep[u]}$
我们发现如果做个变换,即${deep[u]+w[u]==deep[s[i]]}$
与${w[u]-deep[u]==deep[s[i]]-2*deep[lca[i]]}$的话,式子右边是不变的,那是不是可以开一个${cnt}$数组来统计一下子树中等于右式的出发点和结束点的个数呢?
- 考虑使用${cnt[2][600600]}$存储起点/终点的${deep[s[i]]}$/${deep[s[i]]-2*deep[lca[i]]}$等于${j}$的点的数量,保存在数组${cnt[0 or 1][j]}$中,那么当dfs中当前点到${u}$的时候,${ans[u]}$就可以从cnt数组中得到,特殊情况是:${u}$为第${i}$人的lca时可能导致${s[i]}$与${t[i]}$各对${ans[u]}$贡献一次,所以式子如下
- ${ans[u]=cnt[0][dep[u]+w[u]]+cnt[0][w[u]-dep[u]]-重复部分}$。两下标为上面两个等式的左式,需要注意的是${dep[s[i]]-2*dep[lca[i]]}$有可能是负数,下标可以统一+300000处理成正数(否则挂成50分)。其中重复部分可以在消除lca标记的时候判掉。
- 维护cnt数组的方式容易想到就是在${s[i],t[i],lca[i]}$上打标记,若当前点为${s[i]}$就计入${cnt[0][dep[s[i]]]}$,${t[i]}$同理,${lca[i]}$就把两个标记再退出去
- 然而如果对于每颗子树建立一个cnt,再把cnt合并的话,复杂度是n^2的。于是发现搜到以v为根的子树时后,对v贡献的只有两个下标,那么在搜索点v前,传两个参数a,b记录一下原来的cnt两个值,dfs(v,a,b)时${ans[v]-=a+b}$即可
细节
细节很多,想清楚再写。
代码
// uoj261 #include<algorithm> #include<iostream> #include<cstring> #include<cstdlib> #include<cstdio> #include<cmath> #define inf 2147483640 #define LL long long #define Pi acos(-1.0) #define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout) using namespace std; const int T=300000,maxn=300010; struct edge {int to,next;}e[maxn<<1],g[maxn],ct[maxn]; int cnt[2][10000010],h[maxn],hh[maxn],head[maxn]; int deep[maxn],fa[maxn][30],bin[30],Lca[maxn]; int n,m,c1,c2,c3,ans[maxn],cs[maxn],w[maxn],s[maxn],t[maxn]; void linke(int u,int v) { e[++c1].to=v;e[c1].next=head[u];head[u]=c1; e[++c1].to=u;e[c1].next=head[v];head[v]=c1; } void linkg(int u,int v) {g[++c2].to=v;g[c2].next=h[u];h[u]=c2;} void linkt(int u,int v) {ct[++c3].to=v;ct[c3].next=hh[u];hh[u]=c3;} void dfs(int x) { for (int i=1;i<=20;i++) fa[x][i]=fa[fa[x][i-1]][i-1]; for (int i=head[x];i;i=e[i].next) if (e[i].to!=fa[x][0]) { fa[e[i].to][0]=x; deep[e[i].to]=deep[x]+1; dfs(e[i].to); } } int lca(int x,int y) { if (deep[x]<deep[y]) swap(x,y); int t=deep[x]-deep[y]; for (int i=0;bin[i]<=t;i++) if (bin[i]&t) x=fa[x][i]; for (int i=20;i>=0;i--) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i]; return x==y ? x : fa[x][0]; } void dfs(int x,int ta,int tb) { for (int j,i=head[x];i;i=e[i].next) if (e[i].to!=fa[x][0]) { j=e[i].to; dfs(j,cnt[0][w[j]+deep[j]+T],cnt[1][w[j]-deep[j]+T]); } cnt[0][deep[x]+T]+=cs[x]; for (int j,i=hh[x];i;i=ct[i].next) { j=ct[i].to; cnt[1][deep[s[j]]-deep[Lca[j]]*2+T]++; } ans[x]+=cnt[0][w[x]+deep[x]+T]+cnt[1][w[x]-deep[x]+T]-ta-tb; for (int j,i=h[x];i;i=g[i].next) { j=g[i].to; if (w[x]+deep[x]==deep[s[j]]) ans[x]--; cnt[0][deep[s[j]]+T]--;cnt[1][deep[s[i]]-2*deep[x]+T]--; } } int main() { bin[0]=1;for (int i=1;i<=20;i++) bin[i]=bin[i-1]<<1; scanf("%d%d",&n,&m); for (int u,v,i=1;i<n;i++) { scanf("%d%d",&u,&v); linke(u,v); } dfs(1); for (int i=1;i<=n;i++) scanf("%d",&w[i]); for (int i=1;i<=m;i++) { scanf("%d%d",&s[i],&t[i]); Lca[i]=lca(s[i],t[i]);cs[s[i]]++; linkg(Lca[i],i); linkt(t[i],i); } dfs(1,0,0); for (int i=1;i<=n;i++) printf("%d ",ans[i]); return 0; }
以上是关于uoj261 天天爱跑步的主要内容,如果未能解决你的问题,请参考以下文章