首页 > 代码库 > CodeForces 362E Petya and Pipes
CodeForces 362E Petya and Pipes
Petya and Pipes
64-bit integer IO format: %I64d Java class name: (Any)
The Berland capital has n water tanks numbered from 1 to n. These tanks are connected by unidirectional pipes in some manner. Any pair of water tanks is connected by at most one pipe in each direction. Each pipe has a strictly positive integer width. Width determines the number of liters of water per a unit of time this pipe can transport. The water goes to the city from the main water tank (its number is 1). The water must go through some pipe path and get to the sewer tank with cleaning system (its number is n).
Petya wants to increase the width of some subset of pipes by at most k units in total so that the width of each pipe remains integer. Help him determine the maximum amount of water that can be transmitted per a unit of time from the main tank to the sewer tank after such operation is completed.
Input
The first line contains two space-separated integers n and k (2 ≤ n ≤ 50, 0 ≤ k ≤ 1000). Then follow n lines, each line contains n integers separated by single spaces. The i + 1-th row and j-th column contain number cij — the width of the pipe that goes from tank i to tank j (0 ≤ cij ≤ 106, cii = 0). If cij = 0, then there is no pipe from tank i to tank j.
Output
Print a single integer — the maximum amount of water that can be transmitted from the main tank to the sewer tank per a unit of time.
Sample Input
5 7
0 1 0 2 0
0 0 4 10 0
0 0 0 0 5
0 0 0 0 10
0 0 0 0 0
10
5 10
0 1 0 0 0
0 0 2 0 0
0 0 0 3 0
0 0 0 0 4
100 0 0 0 0
5
Hint
In the first test Petya can increase width of the pipe that goes from the 1st to the 2nd water tank by 7 units.
In the second test Petya can increase width of the pipe that goes from the 1st to the 2nd water tank by 4 units, from the 2nd to the 3rd water tank by 3 units, from the 3rd to the 4th water tank by 2 units and from the 4th to 5th water tank by 1 unit.
Source
1 #include <iostream> 2 #include <cstdio> 3 #include <cstring> 4 #include <cmath> 5 #include <algorithm> 6 #include <climits> 7 #include <vector> 8 #include <queue> 9 #include <cstdlib>10 #include <string>11 #include <set>12 #include <stack>13 #define LL long long14 #define pii pair<int,int>15 #define INF 0x3f3f3f3f16 using namespace std;17 const int maxn = 110;18 struct arc {19 int to,flow,cost,next;20 arc(int x = 0,int y = 0,int z = 0,int nxt = -1) {21 to = x;22 flow = y;23 cost = z;24 next = nxt;25 }26 };27 arc e[10000];28 int head[maxn],d[maxn],p[maxn],S,T,n,k,tot;29 bool in[maxn];30 void add(int u,int v,int flow,int cost) {31 e[tot] = arc(v,flow,cost,head[u]);32 head[u] = tot++;33 e[tot] = arc(u,0,-cost,head[v]);34 head[v] = tot++;35 }36 bool spfa() {37 queue<int>q;38 for(int i = 0; i <= n; ++i) {39 d[i] = INF;40 p[i] = -1;41 in[i] = false;42 }43 d[S] = 0;44 q.push(S);45 while(!q.empty()) {46 int u = q.front();47 q.pop();48 for(int i = head[u]; ~i; i = e[i].next) {49 if(e[i].flow && d[e[i].to] > d[u] + e[i].cost) {50 d[e[i].to] = d[u] + e[i].cost;51 p[e[i].to] = i;52 q.push(e[i].to);53 }54 }55 }56 return p[T] > -1;57 }58 int solve() {59 int cost = 0,flow = 0;60 while(spfa()) {61 int theMin = INF;62 for(int i = p[T]; ~i; i = p[e[i^1].to])63 theMin = min(theMin,e[i].flow);64 if(cost + d[T]*theMin > k) return flow + (k - cost)/d[T];65 flow += theMin;66 cost += d[T]*theMin;67 for(int i = p[T]; ~i; i = p[e[i^1].to]) {68 e[i].flow -= theMin;69 e[i^1].flow += theMin;70 }71 }72 return flow;73 }74 int main() {75 while(~scanf("%d %d",&n,&k)) {76 memset(head,-1,sizeof(head));77 S = 0;78 T = n-1;79 int tmp;80 for(int i = tot = 0; i < n; ++i)81 for(int j = 0; j < n; ++j) {82 scanf("%d",&tmp);83 if(tmp) {84 add(i,j,tmp,0);85 add(i,j,k,1);86 }87 }88 printf("%d\n",solve());89 }90 return 0;91 }
上面代码可以AC,但是忘记标记入队点了。。。。。。^_^..
1 #include <iostream> 2 #include <cstdio> 3 #include <cstring> 4 #include <cmath> 5 #include <algorithm> 6 #include <climits> 7 #include <vector> 8 #include <queue> 9 #include <cstdlib>10 #include <string>11 #include <set>12 #include <stack>13 #define LL long long14 #define pii pair<int,int>15 #define INF 0x3f3f3f3f16 using namespace std;17 const int maxn = 110;18 struct arc {19 int to,flow,cost,next;20 arc(int x = 0,int y = 0,int z = 0,int nxt = -1) {21 to = x;22 flow = y;23 cost = z;24 next = nxt;25 }26 };27 arc e[10000];28 int head[maxn],d[maxn],p[maxn],S,T,n,k,tot;29 bool in[maxn];30 void add(int u,int v,int flow,int cost) {31 e[tot] = arc(v,flow,cost,head[u]);32 head[u] = tot++;33 e[tot] = arc(u,0,-cost,head[v]);34 head[v] = tot++;35 }36 bool spfa() {37 queue<int>q;38 for(int i = 0; i <= n; ++i) {39 d[i] = INF;40 p[i] = -1;41 in[i] = false;42 }43 d[S] = 0;44 q.push(S);45 while(!q.empty()) {46 int u = q.front();47 q.pop();48 in[u] = false;49 for(int i = head[u]; ~i; i = e[i].next) {50 if(e[i].flow && d[e[i].to] > d[u] + e[i].cost) {51 d[e[i].to] = d[u] + e[i].cost;52 p[e[i].to] = i;53 if(!in[e[i].to]){54 in[e[i].to] = true;55 q.push(e[i].to);56 }57 }58 }59 }60 return p[T] > -1;61 }62 int solve() {63 int cost = 0,flow = 0;64 while(spfa()) {65 int theMin = INF;66 for(int i = p[T]; ~i; i = p[e[i^1].to])67 theMin = min(theMin,e[i].flow);68 if(cost + d[T]*theMin > k) return flow + (k - cost)/d[T];69 flow += theMin;70 cost += d[T]*theMin;71 for(int i = p[T]; ~i; i = p[e[i^1].to]) {72 e[i].flow -= theMin;73 e[i^1].flow += theMin;74 }75 }76 return flow;77 }78 int main() {79 while(~scanf("%d %d",&n,&k)) {80 memset(head,-1,sizeof(head));81 S = 0;82 T = n-1;83 int tmp;84 for(int i = tot = 0; i < n; ++i)85 for(int j = 0; j < n; ++j) {86 scanf("%d",&tmp);87 if(tmp) {88 add(i,j,tmp,0);89 add(i,j,k,1);90 }91 }92 printf("%d\n",solve());93 }94 return 0;95 }
CodeForces 362E Petya and Pipes