bzoj 4390:[Usaco2015 dec]Max Flow（luogu 3128）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_42369449/article/details/82810944

算法：树上差分+倍增LCA

难度：NOIP-

  好吧，这道题就是 树上差分模板&&倍增LCA模板。结束了...

代码如下：

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <queue>
#include <algorithm>
#define ll long long
#define N 100005
using namespace std;
struct node
{
    int next;
    int to;
}edge[N<<1];
int head[N],dis[N],vis[N];
int cnt=1;
int dep[N],f[N][35];
void init()
{
    memset(head,-1,sizeof(head));
    cnt=1;
}
void add(int u,int v)
{
    edge[cnt].next=head[u];
    edge[cnt].to=v;
    head[u]=cnt++;
}
void dfs(int rx,int fx,int de)
{
    dep[rx]=de;
    f[rx][0]=fx;
    for(int i = head[rx];i != -1;i=edge[i].next)
    {
        int to=edge[i].to;
        if(to==fx) continue;
        dfs(to,rx,de+1);
    }
}
int LCA(int x,int y)
{
    if(dep[x]>dep[y])
    {
        swap(x,y);
    }
    for(int i = 25;i >= 0;i--)
    {
        if(dep[f[y][i]]>=dep[x])
        {
            y=f[y][i];
        }
    }
    if(x==y) return x;
    int ret;
    for(int i = 25;i >= 0;i--)
    {
        if(f[y][i]!=f[x][i])
        {
            x=f[x][i];
            y=f[y][i];
        }else
        {
            ret=f[x][i];
        }
    }
    return f[x][0];
}
int ans=0;
void get_ans(int rt,int fa)
{
    for (int i = head[rt];i != -1;i=edge[i].next)
    {
        int to=edge[i].to;
        if (to==fa) continue;
        get_ans(to,rt);
        vis[rt]+=vis[to];
    }
    ans=max(ans,vis[rt]);
}
int main()
{
    init();
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i = 1;i < n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(1,1,1);//刚刚把dfs的位置放错了，居然苟过了样例 ......
    for(int i = 1;i <= 25;i++)
    {
        for(int j = 1;j <= n;j++)
        {
            f[j][i]=f[f[j][i-1]][i-1];
        }
    }
    //刚刚写到了这里... 
    while(m--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        int lac=LCA(a,b);
        ++vis[a],++vis[b],--vis[lac],--vis[f[lac][0]];
    }
    get_ans(1,-1);
    printf("%d\n",ans);
    return 0 ;
}