BZOJ 4873 [Shoi2017]壽司餐廳 | 網絡流 最大權閉合子圖
阿新 • • 發佈:2017-12-28
blog oid void cap 題解 template ret n) enter
鏈接
BZOJ 4873
題解
當年的省選題……還記得蒟蒻的我Day1 20分滾粗……
這道題是個最大權閉合子圖的套路題。嚴重懷疑出題人就是先畫好了圖然後照著圖編了個3000字的題面。和我喜歡的妹子當年給別人寫的情書一樣長……
最大權閉合子圖
最大權閉合子圖問題:一個有向圖中,每個點帶有一個權值(有正有負),有向邊\(u \to v\)表示選\(u\)必須選\(v\),選出一些點使權值和最大,問權值和最大是多少。
最大權閉合子圖的解法:網絡流建圖,對於每個點\(u\),設權值為\(w_u\),如果\(w_u\)為正,則連邊\((S, u, w_u)\);若為負,連邊\((u, T, |w_u|)\)
。對於原圖中每條有向邊\(u \to v\),連邊\((u, v, +\infty)\)。答案即為所有正點權之和減去網絡流的最小割。
這道題的依賴關系很明確:
- 選擇了一個區間\([i, j](i < j)\),則必須選擇它的兩個子區間\([i, j - 1], [i + 1, j]\)。
- 選擇了一個區間\([i, i]\),則必須選擇裏面的唯一壽司\(i\)。
- 選擇了一個壽司\(i\),則必須選擇它的代號\(a_i\)。
點的權值也很明確:
- “代號”\(i\)的權值為\(- m * i ^ 2\)。
- “代號”為\(i\)的壽司的權值為\(- i\)。
- 區間\([i, j]\)的權值就是輸入數據中的“美味度”。
有了權值和依賴關系,按照上面的最大權閉合子圖的建圖套路無腦建圖即可。
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long ll;
#define enter putchar('\n')
#define space putchar(' ')
template <class T>
void read(T &x){
char c;
bool op = 0;
while(c = getchar(), c > '9' || c < '0')
if(c == '-') op = 1;
x = c - '0';
while(c = getchar(), c >= '0' && c <= '9')
x = x * 10 + c - '0';
if(op) x = -x;
}
template <class T>
void write(T x){
if(x < 0) putchar('-'), x = -x;
if(x >= 10) write(x / 10);
putchar('0' + x % 10);
}
const int N = 10005, M = 1000005, INF = 0x3f3f3f3f;
int n, m, src, des, sum;
int ecnt = 1, adj[N], cur[N], dis[N], nxt[M], go[M], cap[M];
int a[105], w[105][105], lst[1005], idx, type_num[1005], sushi_num[105], range_num[105][105];
void ADD(int u, int v, int _cap){
go[++ecnt] = v;
nxt[ecnt] = adj[u];
adj[u] = ecnt;
cap[ecnt] = _cap;
}
void add(int u, int v, int _cap){
ADD(u, v, _cap);
ADD(v, u, 0);
}
bool bfs(){
static int que[N], qr;
for(int i = 1; i <= des; i++)
cur[i] = adj[i], dis[i] = -1;
que[qr = 1] = src, dis[src] = 1;
for(int ql = 1; ql <= qr; ql++){
int u = que[ql];
for(int e = adj[u], v; e; e = nxt[e])
if(cap[e] && dis[v = go[e]] == -1){
dis[v] = dis[u] + 1, que[++qr] = v;
if(v == des) return 1;
}
}
return 0;
}
int dfs(int u, int flow){
if(u == des) return flow;
int ret = 0, delta;
for(int &e = cur[u], v; e; e = nxt[e])
if(cap[e] && dis[v = go[e]] == dis[u] + 1){
delta = dfs(v, min(cap[e], flow - ret));
if(delta){
cap[e] -= delta;
cap[e ^ 1] += delta;
ret += delta;
if(ret == flow) return ret;
}
}
dis[u] = -1;
return ret;
}
int maxflow(){
int ret = 0;
while(bfs()) ret += dfs(src, INF);
return ret;
}
int main(){
read(n), read(m);
for(int i = 1; i <= n; i++)
read(a[i]), lst[++idx] = a[i];
sort(lst + 1, lst + idx + 1);
int tmp = idx = unique(lst + 1, lst + idx + 1) - lst - 1;
for(int i = 1; i <= lst[idx]; i++)
type_num[i] = lower_bound(lst + 1, lst + idx + 1, i) - lst;
for(int i = 1; i <= n; i++)
sushi_num[i] = ++idx;
for(int i = 1; i <= n; i++)
for(int j = i; j <= n; j++){
range_num[i][j] = ++idx;
read(w[i][j]);
}
src = ++idx, des = ++idx;
for(int i = 1; i <= n; i++){
add(sushi_num[i], des, a[i]);
add(sushi_num[i], type_num[a[i]], INF);
add(range_num[i][i], sushi_num[i], INF);
}
for(int i = 1; i <= n; i++)
for(int j = i; j <= n; j++){
if(j > i){
add(range_num[i][j], range_num[i][j - 1], INF);
add(range_num[i][j], range_num[i + 1][j], INF);
}
if(w[i][j] > 0) add(src, range_num[i][j], w[i][j]), sum += w[i][j];
else add(range_num[i][j], des, -w[i][j]);
}
for(int i = 1; i <= tmp; i++)
add(i, des, lst[i] * lst[i] * m);
write(sum - maxflow()), enter;
return 0;
}
BZOJ 4873 [Shoi2017]壽司餐廳 | 網絡流 最大權閉合子圖