似乎好久没发过博客了哎…… 好不容易有空写道题了,发个博客骗点访问量 传送门
出题人的做法是$O(n\log^2 n)$的,然而我们有$O(n\log n)$的做法 定义$f_{i,j}$表示$i$的从下往上数第$2^j-1$个祖先($f_{i,0}=i$),显然有$f_{i,j}=parent_{f_{f_{i,j-1},j-1}}$(用方括号来写就是$f[i][j]=parent[f[f[i][j-1]][j-1]]$),可以$O(n\log n)$预处理出来 考虑按位统计答案,在统计每位的贡献时可以对每个点枚举它的祖先并考虑这个祖先作为LCA的方案数 通过观察可以发现在统计第$j$位的贡献时离$i$最近的一段有贡献祖先一定是$f_{i,j}$以上(不包括$f_{i,j}$),$f_{i,j+1}$以下(包括$f_{i,j+1}$)这一条链的结点,而以它们作为LCA的方案数显然就是$size_{f_{i,j+1}}-size_{f_{i,j}}$($size_i$表示$i$的子树大小) 至于离$i$并非最近的那几段,容易看出和$parent_{f_{i,j}}$的贡献是一样的,直接再加上$parent_{f_{i,j}}$的贡献即可 答案显然是每一位的每个结点的贡献之和,总共枚举$O(\log n)$位,每位$O(n)$,算上预处理,总复杂度$O(n\log n)$
(好困……写的有点仓促,见谅)
1#include<cstdio> 2#include<cstring> 3#include<algorithm> 4#include<vector> 5using namespace std; 6const int maxn=100005; 7void dfs(int); 8vector<int>G[maxn]; 9int n,p[maxn],d[maxn],f[maxn][23],size[maxn],a[maxn],cnt=0,w[maxn]; 10int main(){ 11 scanf("%d",&n); 12 for(int i=1,x,y;i<n;i++){ 13 scanf("%d%d",&x,&y); 14 G[x].push_back(y); 15 G[y].push_back(x); 16 } 17 p[1]=1; 18 dfs(1); 19 for(int i=1;i<=n;i++){ 20 f[i][0]=i; 21 f[i][1]=p[i]; 22 } 23 for(int j=2;j<=21;j++)for(int i=1;i<=n;i++)f[i][j]=p[f[f[i][j-1]][j-1]]; 24 long long ans=0; 25 for(int j=0;j<=20;j++){ 26 memset(w,0,sizeof(w)); 27 for(int i=1;i<=n;i++){ 28 int x=a[i]; 29 w[x]=size[f[x][j+1]]-size[f[x][j]]; 30 if(d[x]>(1<<(j+1)))w[x]+=w[p[f[x][j+1]]]; 31 ans+=w[x]; 32 } 33 } 34 printf("%lld",ans); 35 return 0; 36} 37void dfs(int x){ 38 a[++cnt]=x; 39 size[x]=1; 40 d[x]=d[p[x]]+1; 41 for(int i=0;i<(int)G[x].size();i++)if(G[x][i]!=p[x]){ 42 p[G[x][i]]=x; 43 dfs(G[x][i]); 44 size[x]+=size[G[x][i]]; 45 } 46}
mdzz……下次放假又不知是啥时候的事儿了……垃圾hz毁我青春 不过没关系……一个月后,又是一条好汉!