// UVa 10983 - Buy one, get the rest free
// binary search + max flow
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <queue>
#include <cmath>
using namespace std;
struct edge {
int from, to, cap, cost, day;
};
typedef long long LL;
struct Edge {
int from, to, cap, flow, index;
Edge(int from, int to, int cap, int flow, int index) :
from(from), to(to), cap(cap), flow(flow), index(index) {
}
};
struct PushRelabel {
int N;
vector<vector<Edge> > G;
vector<LL> excess;
vector<int> dist, active, count;
queue<int> Q;
PushRelabel(int N) :
N(N), G(N), excess(N), dist(N), active(N), count(2 * N) {
}
void AddEdge(int from, int to, int cap) {
G[from].push_back(Edge(from, to, cap, 0, G[to].size()));
if (from == to)
G[from].back().index++;
G[to].push_back(Edge(to, from, 0, 0, G[from].size() - 1));
}
void Enqueue(int v) {
if (!active[v] && excess[v] > 0) {
active[v] = true;
Q.push(v);
}
}
void Push(Edge &e) {
int amt = int(min(excess[e.from], LL(e.cap - e.flow)));
if (dist[e.from] <= dist[e.to] || amt == 0)
return;
e.flow += amt;
G[e.to][e.index].flow -= amt;
excess[e.to] += amt;
excess[e.from] -= amt;
Enqueue(e.to);
}
void Gap(int k) {
for (int v = 0; v < N; v++) {
if (dist[v] < k)
continue;
count[dist[v]]--;
dist[v] = max(dist[v], N + 1);
count[dist[v]]++;
Enqueue(v);
}
}
void Relabel(int v) {
count[dist[v]]--;
dist[v] = 2 * N;
for (int i = 0; i < G[v].size(); i++)
if (G[v][i].cap - G[v][i].flow > 0)
dist[v] = min(dist[v], dist[G[v][i].to] + 1);
count[dist[v]]++;
Enqueue(v);
}
void Discharge(int v) {
for (int i = 0; excess[v] > 0 && i < G[v].size(); i++)
Push(G[v][i]);
if (excess[v] > 0) {
if (count[dist[v]] == 1)
Gap(dist[v]);
else
Relabel(v);
}
}
LL GetMaxFlow(int s, int t) {
count[0] = N - 1;
count[N] = 1;
dist[s] = N;
active[s] = active[t] = true;
for (int i = 0; i < G[s].size(); i++) {
excess[s] += G[s][i].cap;
Push(G[s][i]);
}
while (!Q.empty()) {
int v = Q.front();
Q.pop();
active[v] = false;
Discharge(v);
}
LL totflow = 0;
for (int i = 0; i < G[s].size(); i++)
totflow += G[s][i].flow;
return totflow;
}
};
bool operator <(const edge & x, const edge & y) {
return x.day < y.day;
}
edge edges[1000];
int z[30];
int main() {
int tt;
scanf("%d", &tt);
for (int t = 1; t <= tt; t++) {
int n, d, m;
scanf("%d%d%d", &n, &d, &m);
vector<int> search_values;
for (int i = 0; i < m; i++) {
scanf("%d%d%d%d%d", &edges[i].from, &edges[i].to, &edges[i].cap,
&edges[i].cost, &edges[i].day);
edges[i].from--;
edges[i].to--;
search_values.push_back(edges[i].cost);
}
int visitors = 0;
for (int i = 0; i < n; i++) {
scanf("%d", &z[i]);
visitors += z[i];
}
sort(edges, edges + m);
search_values.push_back(0);
sort(search_values.begin(), search_values.end());
int ini = 0, fin = search_values.size() - 1;
int sol = -1;
while (ini <= fin) {
int mid = (ini + fin) / 2;
// create graph
PushRelabel flowing((d + 1) * n + 2);
for (int i = 0; i < m; i++)
if (edges[i].cost <= search_values[mid])
flowing.AddEdge(edges[i].day * n + edges[i].from,
(edges[i].day + 1) * n + edges[i].to, edges[i].cap);
for (int i = 0; i < n; i++)
flowing.AddEdge((d + 1) * n, i, z[i]);
flowing.AddEdge(d * n + n - 1, (d + 1) * n + 1, visitors);
for (int i = 0; i < n; i++)
for (int j = 0; j < d; j++)
flowing.AddEdge(j * n + i, (j + 1) * n + i, visitors);
if (visitors == flowing.GetMaxFlow((d + 1) * n, (d + 1) * n + 1)) {
sol = search_values[mid];
fin = mid - 1;
} else
ini = mid + 1;
}
if (sol == -1)
printf("Case #%d: Impossible\n", t);
else
printf("Case #%d: %d\n", t, sol);
}
return 0;
}
Monday, May 4, 2015
UVa 10983 - Buy one, get the rest free
Subscribe to:
Post Comments (Atom)
No comments:
Post a Comment