bzoj3127/3697 [Usaco2013 Open]Yin and Yang

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=3127

http://www.lydsy.com/JudgeOnline/problem.php?id=3697

【题解】

点分治。

f[i,0/1]表示前面一坨路径和为i，是否存在休息站。

分类讨论：休息站在点分的地方，休息站在前面子树，休息站在当前子树

子树合并的时候顺便算一发贡献即可。

# include <stdio.h>
# include <string.h>
# include <algorithm>
// # include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
const int M = 5e5 + ;
const int mod = 1e9+;

# define RG register
# define ST static

int n; ll ans = ;
int head[M], nxt[M], to[M], w[M], tot=;
bool vis[M]; 

inline void add(int u, int v, int _w) {
    ++tot; nxt[tot] = head[u]; head[u] = tot; to[tot] = v; w[tot] = _w;
}
inline void adde(int u, int v, int _w) {
    add(u, v, _w); add(v, u, _w);
}

int sz[M], mx[M];
inline void dfsSize(int x, int fa = ) {
    sz[x] = ; mx[x] = ;
    for (int i=head[x]; i; i=nxt[i]) {
        if(to[i] == fa || vis[to[i]]) continue;
        dfsSize(to[i], x);
        sz[x] += sz[to[i]];
        if(sz[to[i]] > mx[x]) mx[x] = sz[to[i]];
    }
}

int centre = , mi;
inline void dfsCentre(int x, int tp, int fa = ) {
    if(sz[tp] - sz[x] > mx[x]) mx[x] = sz[tp] - sz[x];
    if(mx[x] < mi) mi = mx[x], centre = x;
    for (int i=head[x]; i; i=nxt[i]) {
        if(to[i] == fa || vis[to[i]]) continue;
        dfsCentre(to[i], tp, x);
    }
}

int t[M];
ll f[M][], g[M][];
int mxd, vmin = 1e9, vmax = ;
// g: cur tree, f: all
inline void dfsAns(int x, int v, int fa) {
    if(t[v]) g[v][] ++;
    else g[v][] ++;
    t[v] ++;
    vmin = min(vmin, v), vmax = max(vmax, v);
    for (int i=head[x]; i; i=nxt[i]) {
        if(to[i] == fa || vis[to[i]]) continue;
        dfsAns(to[i], v+w[i], x);
    }
    t[v] --;
}

inline void calcAns(int x) {
    int Vmin = 1e7, Vmax = ;
    f[n][] = ;
    for (int i=head[x]; i; i=nxt[i]) {
        if(vis[to[i]]) continue;
        vmin = 1e7, vmax = ;
        dfsAns(to[i], n+w[i], x);
        Vmin = min(Vmin, vmin), Vmax = max(Vmax, vmax);
        ans += (f[n][]-) * g[n][]; //啥都不动，没有1，是没有意义的，为了计数方便，前面先赋值1.
        for (int j=vmin-n; j<=vmax-n; ++j)
            // 将g合并到f上，统计答案
            ans += f[n-j][] * g[n+j][] + f[n-j][] * g[n+j][] + f[n-j][] * g[n+j][];
        for (int j=vmin; j<=vmax; ++j) {
            f[j][] += g[j][];
            f[j][] += g[j][];
            g[j][] = g[j][] = ;
        }
    }
    for (int j=Vmin; j<=Vmax; ++j) f[j][] = f[j][] = ;
}

inline void dfs(int x) {
    dfsSize(x);
    mi = n; dfsCentre(x, x);
    // cross the centre
    calcAns(centre);
    // ================
    vis[centre] = ;
    for (int i=head[centre]; i; i=nxt[i]) {
        if(!vis[to[i]]) dfs(to[i]);
    }
}

int main() {
    scanf("%d", &n);
    for (int i=; i<n; ++i) {
        int u, v, _w; scanf("%d%d%d", &u, &v, &_w);
        adde(u, v, _w ?  : -);
    }
    dfs();
    printf("%lld\n", ans);
    return ;
}

把0/1记录1/-1会方便很多啊不过注意下标+n