1. 程式人生 > >[USACO15DEC]最大流Max Flow

[USACO15DEC]最大流Max Flow

als ems names lag tarjan space sizeof root 時間復雜度

題目:洛谷P3128。

題目大意:一棵n個點的樹,每次將兩個節點最短路徑所覆蓋的所有節點的流量加1。問你最後流量最大的節點的流量是多少。

解題思路:裸的樹上差分。

對於每次增加流量,我們把兩個節點的流量+1,它們的lca和lca的父親的流量-1。

最後求一遍子樹和,求出來每個節點的子樹和就是該節點實際流量。

最後求最大值即可。

時間復雜度:樹剖、倍增LCA $O(m\log_2 n)$,Tarjan LCA $O(n+m)$。

我采用Tarjan算法,288ms。

C++ Code:

#include<cstdio>
#include<cstring>
#include<vector>
#include<cctype>
#define N 50005
using namespace std;
int n,m,ne=0,nq=0;
bool vis[N],instack[N];
int f[N],head[N],que[N],a[N],zx[N];
#define C c=getchar()
inline int readint(){
    char C;
    bool b=false;
    while(!isdigit(c))b=c==‘-‘,C;
    int d=0;
    while(isdigit(c)){
        d=d*10+c-‘0‘;
        C;
    }
    return (b)?(-d):d;
}
struct query{
    int same,nxt,to,num;
    bool flag;
}q[N<<5];
struct edge{
    int to,nxt;
}e[N<<5];
void add_edge(int x,int y){
    e[++ne].to=y;
    e[ne].nxt=head[x];
    head[x]=ne;
    e[++ne].to=x;
    e[ne].nxt=head[y];
    head[y]=ne;
}
void add_que(int x,int y,int z){
    q[++nq].to=y;
    q[nq].same=nq+1;
    q[nq].num=z;
    q[nq].nxt=que[x];
    que[x]=nq;
    q[++nq].to=x;
    q[nq].same=nq-1;
    q[nq].num=z;
    q[nq].nxt=que[y];
    que[y]=nq;
}
int find(int x){
    if(f[x]==x)return x;
    return f[x]=find(f[x]);
}
void tarjan(int root){
    instack[root]=true;
    for(int i=head[root];i;i=e[i].nxt){
        int v=e[i].to;
        if(instack[v])continue;
        tarjan(v);
        f[v]=root;
        vis[v]=true;
    }
    for(int i=que[root];i;i=q[i].nxt)
    if(vis[q[i].to]&&!q[i].flag){
    	int p=find(q[i].to);
        --a[p];
        --a[zx[p]];
        q[i].flag=q[q[i].same].flag=true;

    }
    instack[root]=false;
}
void dfs(int now){
    instack[now]=true;
    for(int i=head[now];i;i=e[i].nxt)
    if(!instack[e[i].to]){
        dfs(e[i].to);
        a[now]+=a[e[i].to];
    }
}
void dfs2(int now){
    instack[now]=true;
    for(int i=head[now];i;i=e[i].nxt)
    if(!instack[e[i].to]){
        zx[e[i].to]=now;
        dfs2(e[i].to);
    }
}
int main(){
    n=readint(),m=readint();
    memset(vis,0,sizeof(vis));
    memset(instack,0,sizeof instack);
    for(int i=1;i<=n;i++)f[i]=i;
    for(int i=1;i<n;i++){
        int u=readint(),v=readint();
        add_edge(u,v);
    }
    memset(a,0,sizeof a);
    zx[1]=0;
    dfs2(1);
    memset(instack,0,sizeof instack);
    for(int i=1;i<=m;i++){
        int x=readint(),y=readint();
        ++a[x];
        ++a[y];
        if(x!=y)
        add_que(x,y,i);else
        --a[x],--a[zx[x]];
    }
    tarjan(1);
    memset(instack,0,sizeof instack);
    dfs(1);
    int max=0;
    for(int i=1;i<=n;i++)
    if(max<a[i])max=a[i];
    printf("%d\n",max);
    return 0;
}

[USACO15DEC]最大流Max Flow