POJ 1459 Power Network
网络流初步,E-K算法模板
传送门:POJ
题意
n个点,np个发电厂,nc个用电厂,其余是中转站。发电厂只发电,用电厂只用电,中转站不发不用。m条线路,连接各点,有容量上限,无损耗。每组样例,求出所有用电厂最大用电量和。
思路
还是网络流的板子,加一个超级源s,s到每个发电厂有一条边,容量是发电厂发电量,加一个超级汇t,每个用电厂同理到汇连边。跑网络流。
代码
输入有一点恶心,处理一下括号,学到的知识就是%*c可以把这个字符忽略
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <cstdlib>
#include <stack>
using namespace std;
const int MAXN=1007;
const int oo=0x3f3f3f3f;
typedef long long LL;
struct Edge{
int from,to,cap,flow;//cap容量 flow流量
Edge(int u,int v,int c,int f){ from=u;to=v;cap=c;flow=f; }
};
int m;
vector<Edge> edges;//顺序的插入边
vector<int> G[MAXN];//保存边号
int a[MAXN];//起点到i的可改进量
int p[MAXN];//保存当前增广的路径
void init(int n)
{
for(int i=0;i<n;i++) G[i].clear();
edges.clear();
}
void AddEdge(int from,int to,int cap)
{
edges.push_back(Edge(from,to,cap,0));
edges.push_back(Edge(to,from,0,0));
m=edges.size();
G[from].push_back(m-2);
G[to].push_back(m-1);
}
int Maxflow(int s,int t)
{
int flow=0;
for(;;)
{
memset(a,0,sizeof(a));
queue<int> Q;
Q.push(s);
a[s]=oo;
while(!Q.empty())
{
int x=Q.front();Q.pop();
for(int i=0;i<G[x].size();i++)
{
Edge& e=edges[G[x][i]];
if(!a[e.to]&&e.cap>e.flow)
{
p[e.to]=G[x][i];
a[e.to]=min(a[x],e.cap-e.flow);
Q.push(e.to);
}
}
if(a[t]) break;
}
if(!a[t]) break;
for(int u=t;u!=s;u=edges[p[u]].from)
{
edges[p[u]].flow+=a[t];
edges[p[u]^1].flow-=a[t];
}
flow+=a[t];
}
return flow;
}
int main()
{
int n,np,nc,m;
while(scanf("%d%d%d%d",&n,&np,&nc,&m)==4)
{
init(n+2);
for(int i=0;i<m;i++)
{
int aa,bb,cc;
char tt[100];
scanf("%s",tt);
sscanf(tt,"%*c%d%*c%d%*c%d",&aa,&bb,&cc);
AddEdge(aa,bb,cc);
}
int s_s=++n,s_t=++n;
for(int i=0;i<np;i++)
{
int aa,bb;
char tt[100];
scanf("%s",tt);
sscanf(tt,"%*c%d%*c%d",&aa,&bb);
AddEdge(s_s,aa,bb);
}
for(int i=0;i<nc;i++)
{
int aa,bb;
char tt[100];
scanf("%s",tt);
sscanf(tt,"%*c%d%*c%d",&aa,&bb);
AddEdge(aa,s_t,bb);
}
int res=Maxflow(s_s,s_t);
printf("%d\n",res);
}
return 0;
}