[Usaco2015 12月]マックス・フロー

タイトル説明

木を考えるには、Nポイント、すべてのノードに0の重みを持っています。

動作K回、毎回2つの指定点S、T、経路上の全ての点の右側に存在Sが値tに追加されます。

右Kオペレーションの値を指す最大重量が完了した後に出力してください。

入力形式

入力の最初の行は、N及びKを含んでいます

次のN-1行はそれぞれストールxとyの間の配管を記述する（≠Y x）の二つの整数x、yを含みます。

次のK線それぞれはミルクが圧送されてれる経路の終点ストールを記述する二つの整数SSとTを含みます。

出力フォーマット

牛乳の最大量を指定する整数は、納屋の任意失速を通してポンピング。

---

一見チェーン割り木で。しかし、アルゴリズムは、そのような暴力を使用する必要はありませんようです。

私たちは、木の差を使用することができます。どのように違いそれは？パスS->トンのために、我々はここで、sとt + 1、およびポイント右LCA -1、右LCAでのポイントの後、父-1右のポイントを置くことができます。これが完了すると、我々は再び見つけ、プレフィックスツリーは、元のパスに右のポイント値を見つける1に追加されますが、（感情的な理由を理解）の他の部分に影響を与えませんでした。接頭辞と最大の場所が答えです。

Tarjan要求にLCAは、その後、時間が複雑性O（N + M）を行うことができます。

#include<iostream>
#include<cstring>
#include<cstdio>
#define maxn 50001
#define maxm 100001
using namespace std;
 
struct edge{
    int to,next;
    edge(){}
    edge(const int &_to,const int &_next){ to=_to,next=_next; }
}e[maxn<<1];
int head[maxn],k;
 
struct question{
    int to,lca,next; bool flag;
    question(){}
    question(const int &_to,const int &_next,const bool &_flag){ to=_to,next=_next,flag=_flag; }
}q[maxm<<1];
int qhead[maxn],qk;
 
int sum[maxn],fa[maxn],pre[maxn];
bool vis[maxn];
int n,m,ans;
 
inline int read(){
    register int x(0),f(1); register char c(getchar());
    while(c<'0'||'9'<c){ if(c=='-') f=-1; c=getchar(); }
    while('0'<=c&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
    return x*f;
}
inline void add(const int &u,const int &v){ e[k]=edge(v,head[u]),head[u]=k++; }
inline void link(const int &u,const int &v,const bool &f){ q[qk]=question(v,qhead[u],f),qhead[u]=qk++; }
 
int get(int x){ return x==fa[x]?x:fa[x]=get(fa[x]); }
void tarjan(int u){
    vis[u]=true,fa[u]=u;
    for(register int i=head[u];~i;i=e[i].next){
        int v=e[i].to;
        if(vis[v]) continue;
        pre[v]=u,tarjan(v),fa[v]=u;
    }
    for(register int i=qhead[u];~i;i=q[i].next){
        int v=q[i].to;
        if(vis[v]&&!q[i].flag){
            q[i].flag=q[i^1].flag=true,q[i].lca=q[i^1].lca=get(v);
            sum[u]++,sum[v]++,sum[get(v)]--,sum[pre[get(v)]]--;
        }
    }
}
 
void dfs_getans(int u){
    for(register int i=head[u];~i;i=e[i].next){
        int v=e[i].to;
        if(v==pre[u]) continue;
        dfs_getans(v);
        sum[u]+=sum[v];
    }
    ans=max(ans,sum[u]);
}
 
int main(){
    memset(head,-1,sizeof head),memset(qhead,-1,sizeof qhead);
    n=read(),m=read();
    for(register int i=1;i<n;i++){
        int u=read(),v=read();
        add(u,v),add(v,u);
    }
    for(register int i=1;i<=m;i++){
        int u=read(),v=read();
        link(u,v,false),link(v,u,false);
    }
 
    tarjan(1);
    dfs_getans(1);
    printf("%d\n",ans);
    return 0;
}

---

そして、木（ほとんどの時間外）のコードセクションを置きます：

#include<iostream>
#include<cstring>
#include<cstdio>
#define maxn 50001
using namespace std;

struct edge{
    int to,next;
    edge(){}
    edge(const int &_to,const int &_next){
        to=_to,next=_next;
    }
}e[maxn<<1];
int head[maxn],k;

struct node{
    int l,r,c,f;
}t[maxn<<2];

inline int read(){
    register int x(0),f(1); register char c(getchar());
    while(c<'0'||'9'<c){ if(c=='-') f=-1; c=getchar(); }
    while('0'<=c&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
    return x;
}

int size[maxn],dep[maxn],son[maxn],fa[maxn];
int top[maxn],dfn[maxn],id[maxn],tot;
int n,m;

inline void add(const int &u,const int &v){
    e[k]=edge(v,head[u]);
    head[u]=k++;
}

void dfs_getson(int u){
    size[u]=1;
    for(register int i=head[u];~i;i=e[i].next){
        int v=e[i].to;
        if(v==fa[u]) continue;
        fa[v]=u,dep[v]=dep[u]+1;
        dfs_getson(v);
        size[u]+=size[v];
        if(size[v]>size[son[u]]) son[u]=v;
    }
}

void dfs_rewrite(int u,int tp){
    top[u]=tp,dfn[u]=++tot,id[tot]=u;
    if(son[u]) dfs_rewrite(son[u],tp);
    for(register int i=head[u];~i;i=e[i].next){
        int v=e[i].to;
        if(v!=son[u]&&v!=fa[u]) dfs_rewrite(v,v);
    }
}

void build(int d,int l,int r){
    t[d].l=l,t[d].r=r;
    if(l==r) return;
    int mid=l+r>>1;
    build(d<<1,l,mid),build(d<<1|1,mid+1,r);
}

inline void down(const int &d){
    t[d<<1].c+=t[d].f*(t[d<<1].r-t[d<<1].l+1);
    t[d<<1|1].c+=t[d].f*(t[d<<1|1].r-t[d<<1|1].l+1);
    t[d<<1].f+=t[d].f;
    t[d<<1|1].f+=t[d].f;
    t[d].f=0;
}

void change(int d,const int &l,const int &r){
    if(l<=t[d].l&&t[d].r<=r){ t[d].c+=t[d].r-t[d].r+1,t[d].f++; return; }
    if(t[d].f) down(d);
    int mid=t[d].l+t[d].r>>1;
    if(l<=mid) change(d<<1,l,r);
    if(r>mid) change(d<<1|1,l,r);
    t[d].c=t[d<<1].c+t[d<<1|1].c;
}

int find(int d,const int &x){
    if(t[d].l==t[d].r) return t[d].c;
    if(t[d].f) down(d);
    int mid=t[d].l+t[d].r>>1;
    if(x<=mid) return find(d<<1,x);
    else return find(d<<1|1,x);
}

int main(){
    memset(head,-1,sizeof head);
    n=read(),m=read();
    for(register int i=1;i<n;i++){
        int u=read(),v=read();
        add(u,v),add(v,u);
    }
    dfs_getson(1);
    dfs_rewrite(1,1);
    build(1,1,tot);
    
    for(register int i=1;i<=m;i++){
        int u=read(),v=read();
        while(top[u]!=top[v]){
            if(dep[top[u]]>dep[top[v]]) swap(u,v);
            change(1,dfn[top[v]],dfn[v]);
            v=fa[top[v]];
        }
        if(dep[u]>dep[v]) swap(u,v);
        change(1,dfn[u],dfn[v]);
    }
    
    int ans=0;
    for(int i=1;i<=n;i++) ans=max(ans,find(1,dfn[i]));
    printf("%d\n",ans);
    return 0;
}