【USACO 2015 Dec Gold】最大流树链剖分-优快云博客

题目描述

　　给定一棵有N个点的树，所有节点的权值都为0。
有K次操作，每次指定两个点s,t，将s到t路径上所有点的权值都加一。
请输出K次操作完毕后权值最大的那个点的权值。

题目大意

树上路径的点权加一，询问操作完后最大的点权。

数据范围

（2≤N≤50,000）（1≤K≤100,000）

样例输入

5 10
3 4
1 5
4 2
5 4
5 4
5 4
3 5
4 3
4 3
1 3
3 5
5 4
1 5
3 4

样例输出

9

解题思路

良心样例！
树链剖分模板题。。。

代码

#include <bits/stdc++.h>
#define Maxn 50005
using namespace std;
inline int Getint(){int x=0,f=1;char ch=getchar();while('0'>ch||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while('0'<=ch&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}
int h[Maxn],mark[Maxn],dep[Maxn],size[Maxn],son[Maxn],fa[Maxn],top[Maxn],cnt=0;
struct node{int to,next;}e[Maxn*2];
void AddEdge(int x,int y){e[++cnt]=(node){y,h[x]};h[x]=cnt;}
void dfs1(int x,int L,int d){
    size[x]=1;
    dep[x]=d;
    fa[x]=L;
    int Max=0;
    for(int p=h[x];p;p=e[p].next){
        int y=e[p].to;
        if(y==L)continue;
        dfs1(y,x,d+1);
        size[x]+=size[y];
        if(size[y]>Max){
            Max=size[y];
            son[x]=y;
        }
    }
}
void dfs2(int x,int st){
    mark[x]=++cnt;
    top[x]=st;
    if(son[x])dfs2(son[x],st);
    for(int p=h[x];p;p=e[p].next)
        if(e[p].to!=fa[x]&&e[p].to!=son[x])
            dfs2(e[p].to,e[p].to);
}
namespace seg{
    struct node{int L,r,Add,vl;}Tree[Maxn*4];
    void Build(int v,int L,int r){
        Tree[v]=(node){L,r,0,0};
        if(L==r)return;
        Build(2*v,L,(L+r)/2);
        Build(2*v+1,(L+r)/2+1,r);
    }
    void PushDown(int x){
        if(Tree[x].L!=Tree[x].r){
            Tree[2*x].Add+=Tree[x].Add;
            Tree[2*x+1].Add+=Tree[x].Add;
            Tree[2*x].vl+=Tree[x].Add;
            Tree[2*x+1].vl+=Tree[x].Add;
            Tree[x].Add=0;
        }
    }
    void Modify(int v,int L,int r){
        if(r<Tree[v].L||Tree[v].r<L)return;
        if(L<=Tree[v].L&&Tree[v].r<=r){
            Tree[v].Add++;
            Tree[v].vl++;
            return;
        }
        PushDown(v);
        Modify(2*v,L,r);
        Modify(2*v+1,L,r);
    }
    int Ask(int v){
        if(Tree[v].L==Tree[v].r)return Tree[v].vl;
        PushDown(v);
        return max(Ask(2*v),Ask(2*v+1));
    }
}
int main(){
    int n=Getint(),m=Getint();
    for(int i=1;i<n;i++){
        int x=Getint(),y=Getint();
        AddEdge(x,y);
        AddEdge(y,x);
    }
    cnt=0;
    dfs1(1,0,1);
    dfs2(1,1);
    seg::Build(1,1,n);
    while(m--){
        int L=Getint(),r=Getint(),f1=top[L],f2=top[r];
        while(f1!=f2){
            if(dep[f1]<dep[f2])swap(f1,f2),swap(L,r);
            seg::Modify(1,mark[f1],mark[L]);
            L=fa[f1];
            f1=top[L];
        }
        if(dep[L]>dep[r])swap(L,r);
        seg::Modify(1,mark[L],mark[r]);
    }
    cout<<seg::Ask(1)<<"\n";
    return 0;
}