[bzoj3532][Sdoi2014]Lis——拆点最小割+字典序+退流

题目大意

给定序列A,序列中的每一项Ai有删除代价Bi和附加属性Ci。请删除若
干项,使得4的最长上升子序列长度减少至少1,且付出的代价之和最小,并输出方案。
如果有多种方案,请输出将删去项的附加属性排序之后,字典序最小的一种。

题解

首先我们很容易用一个\(\Theta (n^2)\)的算法求出对于每个元素的lis。
考虑以下的建图方式:
由S向f[i]==1的点连边,容量为\(\infty\)
由f[i] = max 向 T 连边, 容量为\(\infty\),
对于每个点,拆为两个点,费用就是B[i]。
为什么这样建图是对的?原因很简单,原问题等价于在新图中找一个点集,使得s到t不再连通,这让我们联想到最小割,考虑到s,t到每个点不计入费用,所以我们设为\(\infty\)
那么第一个问题解决了,关键是第二个问题:怎样找出字典序最小的割?
我们这样考虑:既然是字典序最小,我们把C[i]从小到大排序,依次检查每一条边,如果在剩余网络中它不连通,那么这条边一定是原图的一个最小割,我们把这条边删除,同时从汇点开始退流。具体的,我们可以从汇点向v跑一次最大流,u向原点跑一次最大流,这样一定会退回一些流,使得这条边的影响消失。
这样我们就解决了这个问题,详细过程见(我写了4个小时的)代码。

代码

#include <bits/stdc++.h>
using namespace std;
const int maxn = 10000;
int A[maxn], B[maxn], C[maxn], f[maxn];
int n, s, t, v;
const int inf = 1000000010;
struct edge {
  int from;
  int to;
  int cap;
};
struct haha {
  int id;
  int Ci;
} cc[maxn];
 
inline int read() {
  char c = getchar();
  int f = 1, x = 0;
  while (!isdigit(c)) {
    if (c == '-')
      f = -1;
    c = getchar();
  }
  while (isdigit(c))
    x = x * 10 + c - '0', c = getchar();
  return x * f;
}
 
bool cmp(haha a, haha b) { return a.Ci < b.Ci; }
 
vector<edge> edges;
vector<int> G[maxn];
inline void add_edge(int from, int to, int cap) {
  edges.push_back((edge){from, to, cap});
  edges.push_back((edge){to, from, 0});
  int m = edges.size();
  G[from].push_back(m - 2);
  G[to].push_back(m - 1);
}
inline void reaad() {
  scanf("%d", &n);
  s = 0, t = 2 * n + 1, v = t + 1;
  for (int i = 1; i <= n; i++) {
    A[i] = read();
  }
  for (int i = 0; i < v; i++)
    G[i].clear();
  edges.clear();
  for (int i = 1; i <= n; i++)
    B[i] = read();
  for (int i = 1; i <= n; i++) {
    C[i] = read();
    cc[i].id = i;
    cc[i].Ci = C[i];
  }
  for (int i = 1; i <= n; i++)
    add_edge(i, i + n, B[i]);
  for (int i = 1; i <= n; i++)
    f[i] = 1;
  int ans = 0;
  for (int i = 1; i <= n; i++) {
    for (int j = 1; j < i; j++) {
      if (A[j] < A[i])
        f[i] = max(f[i], f[j] + 1);
    }
 
    ans = max(ans, f[i]);
  }
  for (int i = 1; i <= n; i++) {
    if (ans == f[i]) {
      add_edge(i + n, t, inf);
    }
    if (f[i] == 1)
      add_edge(s, i, inf);
    for (int j = 1; j <= n; j++) {
      if (A[j] < A[i] && j < i && f[i] == f[j] + 1)
        add_edge(j + n, i, inf);
    }
  }
}
int dist[maxn], iter[maxn];
inline void bfs(int s) {
  memset(dist, -1, sizeof(dist));
  dist[s] = 0;
  queue<int> q;
  q.push(s);
  while (!q.empty()) {
    int u = q.front();
    q.pop();
    for (int i = 0; i < G[u].size(); i++) {
      edge &e = edges[G[u][i]];
      if (e.cap > 0 && dist[e.to] == -1) {
        dist[e.to] = dist[u] + 1;
        q.push(e.to);
      }
    }
  }
}
inline int dfs(int s, int t, int flow) {
  if (s == t)
    return flow;
  for (int &i = iter[s]; i < G[s].size(); i++) {
    edge &e = edges[G[s][i]];
    if (e.cap > 0 && dist[e.to] > dist[s]) {
      int d = dfs(e.to, t, min(e.cap, flow));
      if (d > 0) {
        e.cap -= d;
        edges[G[s][i] ^ 1].cap += d;
        return d;
      }
    }
  }
  return 0;
}
inline int dinic(int s, int t) {
  int flow = 0;
  while (1) {
    bfs(s);
    if (dist[t] == -1)
      return flow;
    memset(iter, 0, sizeof(iter));
    int d;
    while (d = dfs(s, t, inf))
      flow += d;
  }
  return flow;
}
int main() {
  // freopen("input", "r", stdin);
  int T;
  scanf("%d", &T);
  while (T--) {
    reaad();
    int ans = dinic(s, t);
    int tot = 0;
    int rec[maxn];
    memset(rec, 0, sizeof(rec));
    sort(cc + 1, cc + 1 + n, cmp);
    for (int i = 1; i <= n; i++) {
      int now = cc[i].id;
      int u = now, v = now + n;
      bfs(u);
      if (dist[v] != -1)
        continue;
      rec[tot++] = now;
      dinic(t, v);
      dinic(u, s);
      edges[(now - 1) * 2].cap = edges[(now - 1) * 2 + 1].cap = 0;
    }
    sort(rec, rec + tot);
    printf("%d %d\n", ans, tot);
    for (int i = 0; i < tot-1; i++) {
      printf("%d ", rec[i]);
    }
    printf("%d\n", rec[tot-1]);
  }
  return 0;
}

posted on 2017-02-21 18:01  蒟蒻konjac  阅读(230)  评论(0编辑  收藏  举报