CF1656E Equal Tree Sums题解

其实这道题不难

首先假设 $1$ 是根节点

我看到这道题第一反应就是直接假设整棵树权值之和是某一个定值，然后再dfs造每一个 $a_x$ 。(其实到这里就可以a了，但是不优雅)

我们发现，由于去掉某一个节点之后，它所有的儿子所在的子树之和都和它上面那一部分之和是一样的，于是我们就有了：$su{m}_{x^{\prime }sson}=su{m}_1-su{m}_x$ 其中($x\ne 1$，下同)

又因为：$a_x=su{m}_x-\sum su{m}_{x^{\prime }sson}$

结合一下就成了$a_x=su{m}_1-de{g}_x\times (su{m}_1-su{m}_x)$

当我们令 $su{m}_1=0$ 时，发现：

• $su{m}_{x^{\prime }sson}=-su{m}_x$
• $a_x=de{g}_x\ast su{m}_x$

我们再令 $su{m}_x=(-1{)}^{de{p}_x}$ ，就可以发现 $a_x=de{g}_x(-1{)}^{de{p}_x}$

不难发现，这个结论在 $x=1$ 时也成立

再手动构造几组小数据后发现构造方法正确，就可以优雅的写代码了，芜湖起飞！

最后献上我其丑无比的代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,ans[N];
vector<int>ed[N];
void dfs(int x,int fa,int k){
  ans[x]=ed[x].size()*k;
  for(int y:ed[x])if(fa!=y)dfs(y,x,-k);
}
void solve(){
  cin>>n;
  for(int i=1;i<=n;i++)ed[i].clear();
  for(int i=1,u,v;i<n;i++){
    cin>>u>>v;
    ed[u].emplace_back(v);
    ed[v].emplace_back(u);
  }
  dfs(1,1,1);
  for(int i=1;i<=n;i++)cout<<ans[i]<<" ";
  cout<<'\n';
}
int main(){
  int t;
  cin>>t;
  while(t--)solve();
}