[CQOI2015]网络吞吐量 洛谷3171 BZOJ3931

题目描述

路由是指通过计算机网络把信息从源地址传输到目的地址的活动，也是计算机网络设计中的重点和难点。网络中实现路由转发的硬件设备称为路由器。为了使数据包最快的到达目的地，路由器需要选择最优的路径转发数据包。例如在常用的路由算法OSPF(开放式最短路径优先)中，路由器会使用经典的Dijkstra算法计算最短路径，然后尽量沿最短路径转发数据包。现在，若已知一个计算机网络中各路由器间的连接情况，以及各个路由器的最大吞吐量（即每秒能转发的数据包数量），假设所有数据包一定沿最短路径转发，试计算从路由器1到路由器n的网络的最大吞吐量。计算中忽略转发及传输的时间开销，不考虑链路的带宽限制，即认为数据包可以瞬间通过网络。路由器1到路由器n作为起点和终点，自身的吞吐量不用考虑，网络上也不存在将1和n直接相连的链路。

输入输出格式

输入格式：

输入文件第一行包含两个空格分开的正整数n和m，分别表示路由器数量和链路的数量。网络中的路由器使用1到n编号。接下来m行，每行包含三个空格分开的正整数a、b和d，表示从路由器a到路由器b存在一条距离为d的双向链路。 接下来n行，每行包含一个正整数c，分别给出每一个路由器的吞吐量。

输出格式：

输出一个整数，为题目所求吞吐量。

代码

spfa+dinic 裸题

#include<queue>
#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
typedef long long LL;
const int maxn=5e5+5;
const LL INF=0x3f3f3f3f3f3f3f3f;
bool vis[maxn];
LL dis[maxn],sum[maxn],w[maxn],flw[maxn];
int N,M,S,T,son[maxn],nxt[maxn],lnk[maxn],tot;
inline int read() {
    int ret=0,f=1;char ch=getchar();
    for (; !isdigit(ch); ch=getchar()) if (ch=='-') f=-f;
    for (; isdigit(ch); ch=getchar()) ret=ret*10+ch-48;
    return ret*f;
}
struct Maxflow {
    int son[maxn],nxt[maxn],lnk[maxn],dep[maxn],tot;
    Maxflow() {memset(lnk,-1,sizeof lnk),tot=0;}
    inline void add_edge(int x,int y,LL z) {
        son[tot]=y,flw[tot]=z,nxt[tot]=lnk[x],lnk[x]=tot++;
        son[tot]=x,flw[tot]=0,nxt[tot]=lnk[y],lnk[y]=tot++;
    }
    bool bfs() {
        queue <int> Q;
        memset(dep,0,sizeof dep);
        dep[S]=1,Q.push(S);
        while (!Q.empty()) {
            int u=Q.front();Q.pop();
            for (int k=lnk[u]; ~k; k=nxt[k]) if (dep[son[k]]==0&&flw[k]>0)
                dep[son[k]]=dep[u]+1,Q.push(son[k]);
        }
        return dep[T];
    }
    LL dfs(int x,LL flow) {
        if (x==T||flow==0) return flow;
        LL cnt=0;
        for (int k=lnk[x]; ~k&&flow; k=nxt[k]) if (dep[son[k]]==dep[x]+1&&flw[k]>0) {
            LL d=dfs(son[k],min(flow,flw[k]));
            if (d<=0) continue;
            cnt+=d,flow-=d,flw[k]-=d,flw[k^1]+=d;
        }
        if (cnt==0) dep[x]=0;
        return cnt;
    }
    LL Dinic() {LL ret=0;while (bfs()) ret+=dfs(S,INF);return ret;}
}MF;
inline void add_edge(LL z,int y,int x) {
    son[++tot]=y,w[tot]=z,nxt[tot]=lnk[x],lnk[x]=tot;
    son[++tot]=x,w[tot]=z,nxt[tot]=lnk[y],lnk[y]=tot;
}
void spfa() {
    deque <int> Q;
    memset(vis,0,sizeof vis);
    memset(dis,0x3f,sizeof dis),dis[S]=0,Q.push_back(S);
    while (!Q.empty()) {
        int u=Q.front();
        Q.pop_front(),vis[u]=0;
        for (int k=lnk[u]; k; k=nxt[k]) if (dis[son[k]]>dis[u]+w[k]) {
            dis[son[k]]=dis[u]+w[k];
            if (!vis[son[k]]) {
                vis[son[k]]=1;
                if (!Q.empty()&&dis[Q.front()]>dis[son[k]]) Q.push_front(son[k]);
                else Q.push_back(son[k]);
            }
        }
    }
}
int main() {
   N=read(),M=read(),S=1,T=N+N;
    for (int i=1; i<=M; i++) add_edge(read(),read(),read());
    for (int i=1; i<=N; i++) sum[i]=read();
    sum[1]=sum[N]=INF,spfa();
    for (int i=1; i<=N; i++) {
        MF.add_edge(i,i+N,sum[i]);
        for (int k=lnk[i]; k; k=nxt[k]) if (dis[son[k]]==dis[i]+w[k])
            MF.add_edge(i+N,son[k],INF);
    }
    printf("%lld\n",MF.Dinic());
    return 0;
}