P3171 [CQOI2015]网络吞吐量

传送门

首先跑一遍最短路,如果一条边满足\(dis[v]=dis[u]+w[i]\),那么这条边就在最短路中,把它加进网络流的图里
然后点的流量限制的话拆点,把每个点拆成两个,中间连边来限制流量
最后跑一遍最大流即可,注意两张图不要弄混掉,还有要开\(long\ long\)\(inf\)也要大一点
以上是题解,以下是粗口
出题人我****,数据范围大你说一声,我头一次见不开\(int\)\(inf\)开小直接爆零的你数据好歹出的有点梯度结果我还以为自己哪里打错了找了半天最后只好去看题解才发现要开\(long\ long\)我****

//minamoto
#include<bits/stdc++.h>
#define R register
#define ll long long
#define inf 1e18
#define fp(i,a,b) for(R int i=a,I=b+1;i<I;++i)
#define fd(i,a,b) for(R int i=a,I=b-1;i>I;--i)
#define go(u) for(R int i=head[u],v=e[i].v;i;i=e[i].nx,v=e[i].v)
#define gg(u) for(int &i=cur[u],v=e[i].v;i;i=e[i].nx,v=e[i].v)
using namespace std;
char buf[1<<21],*p1=buf,*p2=buf;
inline char getc(){return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;}
int read(){
    R int res,f=1;R char ch;
    while((ch=getc())>'9'||ch<'0')(ch=='-')&&(f=-1);
    for(res=ch-'0';(ch=getc())>='0'&&ch<='9';res=res*10+ch-'0');
    return res*f;
}
const int N=1005,M=5e5+5;
struct eg{int v,nx;ll w;};int n,m,S,T;
struct net{
    eg e[M];int head[N],tot=1;
    inline void add(R int u,R int v,R ll w){
        e[++tot]={v,head[u],w},head[u]=tot;
        e[++tot]={u,head[v],0},head[v]=tot;
    }
    int q[N],cur[N],dep[N],h,t,u;
    bool bfs(){
        fp(i,S,T)cur[i]=head[i],dep[i]=-1;
        q[h=t=1]=S,dep[S]=0;
        while(h<=t){
            int u=q[h++];
            go(u)if(e[i].w&&dep[v]<0){
                dep[v]=dep[u]+1,q[++t]=v;
                if(v==T)return true;
            }
        }return false;
    }
    ll dfs(int u,ll lim){
        if(u==T||!lim)return lim;
        ll flow=0,f;
        gg(u)if(dep[v]==dep[u]+1&&(f=dfs(v,min(lim,e[i].w)))){
            flow+=f,lim-=f;
            e[i].w-=f,e[i^1].w+=f;
            if(!lim)break;
        }return flow;
    }
    ll dinic(){ll flow=0;while(bfs())flow+=dfs(S,inf);return flow;}
}TT;
struct dis{
    eg e[M];int head[N],tot=1;
    inline void add(R int u,R int v,R ll w){e[++tot]={v,head[u],w},head[u]=tot; }
    int vis[N],h,t,u,v,w,q[N];ll dis[N];
    void spfa(){
        fp(i,1,n)dis[i]=inf,vis[i]=0;
        dis[1]=0,h=0,t=1,q[0]=1;
        while(h!=t){
            u=q[h++],vis[u]=0;if(h==N)h=0;
            go(u)if(dis[v]>dis[u]+e[i].w){
                dis[v]=dis[u]+e[i].w;
                if(!vis[v]){
                    vis[v]=1,q[t++]=v;
                    if(t==N)t=0;
                }
            }
        }
    }
    void MAIN(){
        n=read(),m=read(),S=0,T=(n<<1)+1;
        fp(i,1,m)u=read(),v=read(),w=read(),add(u,v,w),add(v,u,w);
        spfa();fp(u,1,n)go(u)if(dis[v]==dis[u]+e[i].w)TT.add(u+n,v,inf);
        fp(i,1,n)w=read(),TT.add(i,i+n,w);
        TT.add(1,1+n,inf),TT.add(n,n+n,inf),TT.add(S,1,inf),TT.add(n+n,T,inf);
        printf("%lld\n",TT.dinic());
    }
}G;
signed main(){
//  freopen("testdata.in","r",stdin);
    G.MAIN();return 0;
}

猜你喜欢

转载自www.cnblogs.com/bztMinamoto/p/10079732.html
今日推荐