Equal Tree Sums（构造）

Equal Tree Sums

[Link](Problem - E - Codeforces)

题意

给你一棵$n$个结点的树，请你给每个结点分配权值，使得删除任意一个节点后，所有的连通块的权值和相等。

思路

构造题从小的样例找规律，发现和度数相关，我们可以先进行黑白染色，然后将黑色点的权值赋为正的度数，白色点权值赋为负得度数。这样首先所有的权值和是$0$，任意去除一个点所有和他相连的边对应的连通块的权值会和会变成$1$或者$-1$，因此成立。

Code

#include <bits/stdc++.h>
#define x first
#define y second
#define debug(x) cout<<#x<<":"<<x<<endl;
using namespace std;
typedef long double ld;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
typedef unsigned long long ULL;
const int N = 1e5 + 10, M = 2 * N, INF = 0x3f3f3f3f, mod = 1e9 + 7;
const double eps = 1e-8, pi = acos(-1), inf = 1e20;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
int h[N], e[M], ne[M], w[M], idx;
void add(int a, int b, int v = 0) {
    e[idx] = b, w[idx] = v, ne[idx] = h[a], h[a] = idx ++;
}
int n, m, k;
int a[N];
int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    int T;
    cin >> T;
    while (T -- ) {
        cin >> n;
        vector<int> g[n];
        for (int i = 1; i < n; i ++) {
            int u, v; cin >> u >> v;
            u --, v --;
            g[u].push_back(v), g[v].push_back(u);
        }

        function<void(int, int, int)> dfs = [&](int u, int fa, int c) {
            a[u] = c * g[u].size();
            for (auto v : g[u]) {
                if (v == fa) continue ;
                dfs(v, u, -c);
            }
        };

        dfs(0, -1, 1);
        for (int i = 0; i < n; i ++) cout << a[i] << ' ';
        cout << '\n';
    }
    return 0;
}