300iq Contest 3简要题解

和dcx打了一场300iq contest 3，垫底了。
A：
有两个做法。
第一个做法是考虑对于一个合法的点集$S$，一定存在一个树上的连通块，使得从连通块中任何一点，距离$S$中任一点不超过$\frac{x}{2}$。注意当$x$为奇数时，连通块可能会缩到一条边的中点上。考虑枚举连通块中的点，用“点-边”来统计答案。具体是对某个点（边中点），令$T$为距离它不超过$\frac{x}{2}$的点的集合，这个点会在$S$的连通块中当且仅当$S\subseteq T$。于是给每个点和每条边中点算出对应$T$的大小，那么对答案的贡献用一个FFT卷积即可快速计算。算$T$的大小可以用点分治，时间复杂度$\mathcal{O}(n\log n)$。
另一个做法用了一个结论。令$G=T^x$，那么$G$是弦图，且若将节点按以任意节点为根时的深度排序（深度相同任意）的逆序，是一个完美消除序列。有了这个结论，我们可以考虑枚举$S$在完美消除序列中最后出现的点$X$，则在$X$前面，且在树上距离它不超过$x$的集合$T$满足$T+X$是一个团，即$T$中任选节点，加上$X$均合法。这里同样需要用FFT卷积加速，瓶颈在于前面的计算，视时间时间复杂度为$\mathcal{O}(n{\log }^{2}n)$或$\mathcal{O}(n\log n)$。

#include <bits/stdc++.h>
#define MOD 998244353

using namespace std;

typedef long long ll;

ll pow_mod(ll x,int k) {
   
  ll ans=1;
  while (k) {
   
  	if (k&1) ans=ans*x%MOD;
  	x=x*x%MOD;
  	k>>=1;
  }
  return ans;
}

ll facd[300005],facv[300005];

void pre(int n) {
   
  facd[0]=1;
  for(int i=1;i<=n;i++) facd[i]=facd[i-1]*i%MOD;
  facv[n]=pow_mod(facd[n],MOD-2);
  for(int i=n-1;i>=0;i--) facv[i]=facv[i+1]*(i+1)%MOD;
}

const int Maxn=1<<20;

ll *w[20];

void ntt_init() {
   
  for(int i=2,t=0;i<=Maxn;i<<=1,t++) {
   
  	w[t]=new ll[i>>1];
  	ll wn=pow_mod(3,(MOD-1)/i);
  	w[t][0]=1;
  	for(int j=1;j<(i>>1);j++) w[t][j]=w[t][j-1]*wn%MOD;
  }
}

void rev(ll *p,int len) {
   
  int j=len>>1;
  for(int i=1;i<len-1;i++) {
   
  	if (i<j) swap(p[i],p[j]);
  	int k=len>>1;
  	while (j>=k) {
   
  		j-=k;
  		k>>=1;
	  }
	if (j<k) j+=k;
  }
}

void ntt(ll *p,int len,int check) {
   
  rev(p,len);
  for(int i=2,t=0;i<=len;i<<=1,t++)
    for(int j=0;j<len;j+=i)
      for(int k=j;k<j+(i>>1);k++) {
   
      	ll u=p[k];
      	ll v=w[t][k-j]*p[k+(i>>1)];
      	p[k]=(u+v)%MOD;
      	p[k+(i>>1)]=(u-v)%MOD;
	  }
  if (check==-1) {
   
  	reverse(p+1,p+len);
  	ll nev=pow_mod(len,MOD-2);
  	for(int i=0;i<len;i++) p[i]=(p[i]+MOD)*nev%MOD;
  }
}

vector <int> e[600005];
int n,val[300005];

namespace TDC {
   

int size1[600005],size2[600005];
bool vis[600005];
int sz,rt;

void dfs2(int x,int fa) {
   
  size1[x]=1;size2[x]=0;
  for(int i=0;i<e[x].size();i++)
    if (!vis[e[x][i]]&&e[x][i]!=fa) {
   
    	int u=e[x][i];
    	dfs2(u,x);
    	size1[x]+=size1[u];
    	size2[x]=max(size2[x],size1[u]);
	}
}

void dfs3(int x,int fa) {
   
  for(int i=0;i<e[x].size();i++)
    if (!vis[e[x][i]]&&e[x][i]!=fa) dfs3(e[x][i],x);
  size2[x]=max(size2[x],sz-size1[x]);
  if (size2[x]<size2[rt]) rt=x;
}

int p[600005][25],dep[600005][25];
int* sum1[600005],*sum2[600005];

int sumv[600005],maxd;
int max1[600005],max2[600005];

void dfs4(int x,int fa,int v) {
   
  p[x][++p[x][0]]=v;
  if (fa) dep[x][p[x][0]]=dep[fa][p[fa][0]]+1;
  for(int i=0;i<e[x].size();i++)
    if (!vis[e[x][i]]&&e[x][