[2019沈阳网络赛D题]Dawn-K's water(点分治)

题目链接

题意为求出树上任意点对的距离对3取余的和。

比赛上听到题意就知道是点分治了，但是越写越不对劲，交之前就觉得会T，果不其然T了。修修改改结果队友写了发dp直接就过了Orz。

赛后想了想维护的东西太脑残了，以为像洛谷板子题一样暴力维护就可以，实则被卡死。

赛后的想法是维护距离当前重心的距离对3取余后的距离和以及个数。然后统计的时候枚举两个点的距离取余值，然后统计贡献。

注意下传的时候要消除重复的贡献。

代码丑陋请见谅QAQ

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 const int maxn = 3e4 + ;
 const int mod = 1e9 + ;
 struct node {
     int s, e, w, next;
 }edge[maxn * ];
 int head[maxn], len;
 void add(int s, int e, int w) {
     edge[len].e = e;
     edge[len].w = w;
     edge[len].next = head[s];
     head[s] = len++;
 }
 int n, root, sum;
 int vis[maxn], f[maxn], son[maxn];
 ll ans[], o[], num[];
 void getroot(int x, int fa) {
     son[x] = , f[x] = ;
     for (int i = head[x]; i != -; i = edge[i].next) {
         int y = edge[i].e;
         if (y == fa || vis[y])continue;
         getroot(y, x);
         son[x] += son[y];
         f[x] = max(f[x], son[y]);
     }
     f[x] = max(f[x], sum - son[x]);
     if (f[x] < f[root])root = x;
 }
 void getd(int x, int dis, int fa) {
     o[dis % ]++;
     num[dis % ] += dis;
     for (int i = head[x]; i != -; i = edge[i].next) {
         int y = edge[i].e;
         if (y == fa || vis[y])continue;
         getd(y, (dis + edge[i].w) % mod, x);
     }
 }
 void cal(int x, int val, int add) {
     getd(x, val, );
     for (int i = ; i < ; i++)
         for (int j = ; j < ; j++) {
             ans[(i + j) % ] = (ans[(i + j) % ] + o[i] * num[j] * add % mod + mod) % mod;
             ans[(i + j) % ] = (ans[(i + j) % ] + o[j] * num[i] * add % mod + mod) % mod;
         }
     for (int i = ; i < ; i++)o[i] = num[i] = ;

 }
 void solve(int x) {
     cal(x, , );
     vis[x] = ;
     for (int i = head[x]; i != -; i = edge[i].next) {
         int y = edge[i].e;
         if (vis[y])continue;
         cal(y, edge[i].w, -);
         sum = son[y];
         root = ;
         getroot(y, );
         solve(root);
     }
 }
 int main() {
     while (scanf("%d", &n) != EOF) {
         len = ;
         for (int i = ; i <= n + ; i++)
             vis[i] = , head[i] = -;
         for (int i = ; i < n; i++) {
             int x, y, z;
             scanf("%d%d%d", &x, &y, &z);
             x++, y++;
             add(x, y, z);
             add(y, x, z);
         }
         for (int i = ; i < ; i++)
             ans[i] = ;
         root = , f[] = INT_MAX - ;
         sum = n;
         getroot(, );
         solve(root);
         for (int i = ; i <= ; i++) {
             if (i == )
                 printf("%lld\n", ans[i]);
             else
                 printf("%lld ", ans[i]);
         }
     }
 }