HDU 4862 Jump 任意起点最大权K链覆盖

#include<bits/stdc++.h>
#define reg register
using namespace std;
typedef long long ll;
typedef int JQK;
const int INF = 0x7f7f7f7f;
const int MAXN = 505, MAXM = 13000;
int Head[MAXN], cur[MAXN], to[MAXM << 1], nxt[MAXM << 1], f[MAXM << 1], ed = 1;
int S, T, MAXP, MAXF, pre[MAXN];
JQK lev[MAXN], mono[MAXM << 1];
bool exist[MAXN];
inline void RR(int &x) {
        char c;
        bool sign = false;
        for (c = getchar(); c < ‘0‘ || c > ‘9‘; c = getchar())
                if (c == ‘-‘) {
                        sign = true;
                }
        for (x = 0; c >= ‘0‘ && c <= ‘9‘; c = getchar()) {
                x = x * 10 + c - ‘0‘;
        }
        sign && (x = -x);
}
void addedge(int u, int v, int cap, JQK val) {
        //cout<<u<<" "<<v<<" "<<cap<<" "<<val<<endl;
        to[++ed] = v;
        nxt[ed] = Head[u];
        Head[u] = ed;
        f[ed] = cap;
        mono[ed] = val;
        to[++ed] = u;
        nxt[ed] = Head[v];
        Head[v] = ed;
        f[ed] = 0;
        mono[ed] = -1 * val;
        return ;
}
bool BFS() {
        int u;
        queue<int> q;
        for (int i = 0; i <= MAXP + 1; i++) {
                exist[i] = 0;
                lev[i] = INF;
        }
        lev[S] = pre[S] = 0;
        q.push(S);
        while (q.size()) {
                u = q.front();
                q.pop();
                exist[u] = false;
                for (int i = Head[u]; i; i = nxt[i])
                        if (f[i] && lev[u] + mono[i] < lev[to[i]]) {
                                lev[to[i]] = lev[u] + mono[i];
                                pre[to[i]] = i;
                                if (!exist[to[i]]) {
                                        exist[to[i]] = true;
                                        q.push(to[i]);
                                }
                        }
        }
        for (int i = 0; i <= MAXP + 1; i++) {
                cur[i] = Head[i];
        }
        return lev[T] != INF;
}
JQK Augment() {
        JQK delta = INF;
        for (int i = pre[T]; i; i = pre[to[i ^ 1]])
                if (f[i] < delta) {
                        delta = f[i];
                }
        for (int i = pre[T]; i; i = pre[to[i ^ 1]]) {
                f[i] -= delta;
                f[i ^ 1] += delta;
        }
        MAXF += delta;
        return delta * lev[T];
}
JQK MCMF() {
        JQK ans = 0;
        memset(exist, false, sizeof(exist));
        while (BFS()) {
                ans += Augment();
        }
        return ans;
}
void init(int x123, int y123) {
        MAXF = 0;
        for (int i = 0; i <= MAXP; i++) {
                Head[i] = 0;
        }
        ed = 1;
        S = x123;
        T = y123;
        return ;
}
char f1[15][15];
int ff[15][15];
int num[15][15];
int main() {
        int TNT;
        int n, m, K;
        RR(TNT);
        for (int cas = 1; cas <= TNT; cas++) {
                int sum = 0;
                RR(n), RR(m), RR(K);
                for (int i = 1; i <= n; i++) {
                        scanf("%s", f1[i] + 1);
                        for (int j = 1; j <= m; j++) {
                                ff[i][j] = f1[i][j] - ‘0‘;
                        }
                }
                for (int i = 1; i <= n; i++) {
                        for (int j = 1; j <= m; j++) {
                                num[i][j] = ++sum;
                        }
                }
                MAXP = 2 * sum + 3;
                init(2 * sum + 2, 2 * sum + 3);
                addedge(S, S - 1, K, 0);
                for (int i = 1; i <= n; i++) {
                        for (int j = 1; j <= m; j++) {
                                addedge(S, num[i][j], 1, 0);
                                addedge(sum + num[i][j], T, 1, 0);
                                addedge(S - 1, sum + num[i][j], 1, 0);
                        }
                }
                for (int i = 1; i <= n; i++) {
                        for (int j = 1; j <= m; j++) {
                                int now = ff[i][j];
                                for (int k = i + 1; k <= n; k++) {
                                        int now2 = ff[k][j];
                                        int add = (now == now2) ? now : 0;
                                        addedge(num[i][j], sum + num[k][j], 1, -(add - (k - i - 1)));
                                }
                                for (int k = j + 1; k <= m; k++) {
                                        int now2 = ff[i][k];
                                        int add = (now == now2) ? now : 0;
                                        addedge(num[i][j], sum + num[i][k], 1, -(add - (k - j - 1)));
                                }
                        }
                }
                printf("Case %d : ", cas);
                if (K < min(n, m)) {
                        printf("-1\n");
                        continue;
                }
                int anser = MCMF();
                if (MAXF != sum) {
                        printf("-1\n");
                } else {
                        printf("%d\n", -anser);
                }
        }
        return 0;
}