题目描述
有一棵n个点的树和两个整数p, q,求满足以下条件的四元组(a, b, c, d)的个数:
1.$1\leq a,b,c,d \leq n$
2.点a到点b的经过的边数为p。
3.点c到点d的经过的边数为q。
4.不存在一个点,它既在点a到点b的路径上,又在点c到点d的路径上。
输入描述:
第一行三个整数n,p,q。
接下来n - 1行,每行两个整数u, v,表示树上存在一个连接点u和点v的边。
输出描述:
输出一个整数,表示答案。
示例1
输入
5 2 1
1 2
2 3
3 4
2 5
输出
4
说明
合法的四元组一共有:
(1, 5, 3, 4),
(1, 5, 4, 3),
(5, 1, 3 ,4),
(5, 1, 4, 3)。
示例2
输入
4 1 1
1 2
2 3
3 4
输出
8
备注:
对于前20%的数据,n,p,q≤50。
对于前40%的数据,n,p,q≤200。
对于另外10%的数据,p = 2, q = 2。
对于另外10%的数据,树是一条链。
对于另外10%的数据,树随机生成。
对于所有数据1≤n,p,q≤3000,1≤u,v≤n,保证给出的是一棵合法的树。
分析
我已经弱到连$n^2$枚举路径都不会了
再一次求助Master_Yi
这个题只要理顺了就挺好想的了(说得好像我想得出来似的。
由于不相交的情况不好求,所直接看相交的情况。
找规律可以发现,如果两条路径相交,其中必有一条路径两个端点的lca在另一条路径上
所有可以枚举长度为p的路径,减去在以这条路径上的点为端点lca的长度为q的路径
然后又枚举长度为q的路径,减去在以这条路径上的点为端点lca的长度为p的路径
发现当路径端点lca相同的情况被多算了一次,于是就加回来
那么如何实现呢?
设sq[x],sp[x]分别表示以x为端点lca,长度为q和长度为p的路径条数
枚举路径是$n^2$的,如果不能优化的话,那么我们现在需要的是快速求出一条路径上的sq或sp和
现在要求的是一条路径的和,一个一个找点肯定会T,所以可以预处理一些东西能让我们能够拼凑出答案
如果预处理从根到某个节点x上的路径的sq之和与sp之和,记为ssq[x]与spp[x]
那么以i,j为两端点的路径中sq和sp之和就为ssq[i]+ssq[j]-ssq[lca[i][j]]-ssq[fa[lca[i][j]]]与ssp[i]+ssp[j]-ssp[lca[i][j]]-ssp[fa[lca[i][j]]]
感觉有些与前缀和类似。。。。。。
这样就可以O(1)计算了,总的时间复杂度就为O(n^2)
跟Master_Yi几乎一样的Code
#include<cstdio> #include<algorithm> using namespace std; const int maxn=3005; int fa[maxn],dep[maxn],ori[maxn],f[maxn][maxn],lca[maxn][maxn]; int n,p,q,ecnt,Sp,Sq,v[maxn<<1],nx[maxn<<1],sq[maxn],sp[maxn],vis[maxn],info[maxn]; int find(int x){return !ori[x]?x:ori[x]=find(ori[x]);} void add(int u1,int v1){nx[++ecnt]=info[u1];info[u1]=ecnt;v[ecnt]=v1;} void dfs1(int x,int fa) { f[x][0]=1;vis[x]=1;for(int i=1;i<=n;i++)if(vis[i])lca[x][i]=lca[i][x]=find(i); for(int e=info[x];e;e=nx[e])if(v[e]!=fa) { dfs1(v[e],x); for(int i=0;i<q;i++)sq[x]+=f[x][i]*f[v[e]][q-i-1]; for(int i=0;i<p;i++)sp[x]+=f[x][i]*f[v[e]][p-i-1]; for(int i=0;i<max(p,q);i++)f[x][i+1]+=f[v[e]][i]; } ori[x]=fa;Sq+=sq[x];Sp+=sp[x]; } void dfs2(int x,int f){sq[x]+=sq[f];sp[x]+=sp[f];dep[x]=dep[fa[x]=f]+1;for(int e=info[x];e;e=nx[e])if(v[e]!=f)dfs2(v[e],x);} int main() { scanf("%d%d%d",&n,&p,&q); for(int i=1,u1,v1;i<n;i++)scanf("%d%d",&u1,&v1),add(u1,v1),add(v1,u1); dfs1(1,0);dfs2(1,0);long long ans=p!=q?1ll*Sp*Sq:1ll*Sp*(Sq-1)/2; for(int i=1;i<=n;i++)for(int j=i+1;j<=n;j++) { int len=dep[i]+dep[j]-2*dep[lca[i][j]]; if(len==p&&len==q){ans-=sq[i]+sq[j]-sq[lca[i][j]]-sq[fa[lca[i][j]]]-1;continue;} if(len==p)ans-=sq[i]+sq[j]-sq[lca[i][j]]-sq[fa[lca[i][j]]]; if(len==q)ans-=sp[i]+sp[j]-sp[lca[i][j]]-sp[fa[lca[i][j]]]; } for(int i=1;i<=n;i++) if(p==q)ans+=1ll*(sp[i]-sp[fa[i]])*(sq[i]-sq[fa[i]]-1)/2; else ans+=1ll*(sp[i]-sp[fa[i]])*(sq[i]-sq[fa[i]]); printf("%lld\n",p==q?ans<<3:ans<<2); }