【USACO】 Max Flow

【题目链接】

点击打开链接

【算法】

LCA + 树上差分

【代码】

#include<bits/stdc++.h>

using namespace std;

int i,x,y,N,K,l,maxn;
int anc[][],d[],sum[];
vector<int> E[];

void build(int x,int pre) {
        int i;
        for (i = ; i <= ; i++) anc[x][i] = anc[anc[x][i-]][i-];
        for (i = ; i < E[x].size(); i++) {
                if (E[x][i] != pre) {
                        anc[E[x][i]][] = x;
                        d[E[x][i]] = d[x] + ;
                        build(E[x][i],x);
                }
        }
}

int lca(int x,int y) {
        int i,f;
        if (d[x] > d[y]) swap(x,y);
        f = d[y] - d[x];
        for (i = ; i <= ; i++) {
                if (f & ( << i))
                        y = anc[y][i];
        }
        if (x == y) return x;
        for (i = ; i >= ; i--) {
                if (anc[x][i] != anc[y][i]) {
                        x = anc[x][i];
                        y = anc[y][i];
                }
        }
        return anc[x][];
}

void query(int x,int pre) {
        int i;
        for (i = ; i < E[x].size(); i++) {
                if (E[x][i] != pre) {
                        query(E[x][i],x);
                        sum[x] += sum[E[x][i]];
                }
        }
        maxn = max(maxn,sum[x]);
}

int main() {

        cin >> N >> K;
        for (i = ; i < N; i++) {
                cin >> x >> y;
                E[x].push_back(y);
                E[y].push_back(x);
        }

        build(,);

        for (i = ; i <= K; i++) {
                cin >> x >> y;
                l = lca(x,y);
                sum[x]++; sum[y]++;
                sum[l]--;
                if (anc[l][]) sum[anc[l][]]--;
        }

        query(,);

        cout<< maxn << endl;

        return ;
}