洛谷 P3171 [CQOI2015]网络吞吐量

题目描述

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

输入输出格式

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

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

输入输出样例

输入样例#1：
7 10
1 2 2
1 5 2
2 4 1
2 3 3
3 7 1
4 5 4
4 3 1
4 6 1
5 6 2
6 7 1
1
100
20
50
20
60
1
输出样例#1：
70
说明

对于100%的数据，n<=500，m<=100000，d,c<=10^9

---

【分析】
题面=题解系列
先求一遍spfa，然后把不是最短路的边删掉…然后拆点限制容量跑一边dinic…

---

【代码】

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define ll long long
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const ll inf=1e16;
const int mxn=505;
queue <ll> q;
ll n,m,s,t,cnt;
ll D[mxn],dis[mxn];
ll head[mxn],vis[mxn],x[mxn],y[mxn],d[mxn];
struct edge {int to,next;ll flow,d;} f[200005]; 
inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    return x*f;
}
inline void ADD(int u,int v,ll d)
{
    f[++cnt].to=v,f[cnt].next=head[u],f[cnt].d=d,head[u]=cnt;
}
inline void add2(int u,int v,ll flow)
{
    f[++cnt].to=v,f[cnt].next=head[u],f[cnt].flow=flow,head[u]=cnt;
    f[++cnt].to=u,f[cnt].next=head[v],f[cnt].flow=0,head[v]=cnt;
}
inline void spfa()
{
    memset(vis,0,sizeof vis);
    memset(D,0x7f,sizeof D);
    D[1]=0;q.push(1);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(ll i=head[u];i;i=f[i].next)
        {
            ll v=f[i].to;ll w=f[i].d;
            if(D[v]>D[u]+w)
            {
                D[v]=D[u]+w;
                if(!vis[v]) q.push(v),vis[v]=1;
            }
        }
    }
}
inline bool bfs()
{
    memset(dis,-1,sizeof dis);
    dis[s]=0,q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(ll i=head[u];i;i=f[i].next)
        {
            int v=f[i].to;ll flow=f[i].flow;
            if(flow>0 && dis[v]==-1)
                dis[v]=dis[u]+1,q.push(v);
        }
    }
    if(dis[t]>0) return 1;
    return 0;
}
inline ll find(ll u,ll low)
{
    ll a,sum=0;
    if(u==t) return low;
    for(int i=head[u];i;i=f[i].next)
    {
        int v=f[i].to;ll flow=f[i].flow;
        if(dis[v]==dis[u]+1 && flow>0 && (a=find(v,min(flow,low-sum))))
        {
            sum+=a;
            f[i].flow-=a;
            if(i&1) f[i+1].flow+=a;
            else f[i-1].flow+=a;
        }
    }
    if(!sum) dis[u]=-1;
    return sum;
}
int main()
{
//  freopen("cqoi15_network.in","r",stdin);
//  freopen("cqoi15_network.out","w",stdout);
    int i,j,u,v,w;ll ans=0;
    n=read(),m=read();
    s=0,t=n+n+1;
    fo(i,1,m)
    {
        x[i]=read(),y[i]=read(),d[i]=read();
        ADD(x[i],y[i],d[i]),ADD(y[i],x[i],d[i]);
    }
    spfa();
    memset(head,0,sizeof head);
    memset(f,0,sizeof f);cnt=0;
    fo(i,1,m)
    {
        if(D[x[i]]==D[y[i]]+d[i]) add2(y[i]+n,x[i],inf);
        if(D[y[i]]==D[x[i]]+d[i]) add2(x[i]+n,y[i],inf);
    }
    fo(i,1,n)
    {
        w=read();
        if(i!=1 && i!=n) add2(i,i+n,w);
        else add2(i,i+n,inf);
    }
    add2(s,1,inf),add2(n+n,t,inf);
    while(bfs())
      ans+=find(s,inf);
    printf("%lld\n",ans);
    return 0;
}