NKOJ【SCOI2008 Day1】天平

你有n个砝码，均为1克，2克或者3克。你并不清楚每个砝码的重量，但你知道其中一些砝码重量的大小关系。你把其中两个砝码A 和B 放在天平的左边，需要另外选出两个砝码放在天平的右边。问：有多少种选法使得天平的左边重(c1)、一样重(c2)、右边重(c3)？（只有结果保证惟一的选法才统计在内）

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <cmath>
#define N 100010
#define inf 1000000000
using namespace std;

int n, B, A, idc;
int ans[5], head[N], vis[N], dis[N], rec[10];
bool flag;

struct Edge {
    int to, nxt, w;
}ed[N];

void adde(int u, int v, int w){
    ed[++idc].to = v; 
    ed[idc].nxt = head[u]; 
    ed[idc].w = w;
    head[u] = idc; 
}

void spfa(int u){
    vis[u] = 1;
    for(int i=head[u]; i; i=ed[i].nxt){
        int v = ed[i].to;
        if(dis[v] > dis[u] + ed[i].w) {
            dis[v] = dis[u] + ed[i].w;
            if( vis[v] ) {
                flag = true; return;
            }
            spfa( v );
            if( flag ) return;
        }
    }
    vis[u] = 0;
}

void init(int c, int d){
    flag = false;
    for(int i=0; i<=n; i++) dis[i] = inf, vis[i] = 0;
    dis[0] = 0;
    rec[0] = idc; rec[1] = head[A]; rec[2] = head[B]; 
    rec[3] = head[c]; rec[4] = head[d]; rec[5] = head[0]; 
}

void recover(int c, int d){
    idc = rec[0]; head[A] = rec[1]; head[B] = rec[2];
    head[c] = rec[3]; head[d] = rec[4]; head[0] = rec[5];
}

bool check1(int c, int d, int vd, int va){//<
    init(c, d);
    int t = vd - va - 1;
    adde(c, B, t);
    adde(0, d, vd); adde(d, 0, -vd);
    adde(0, A, va); adde(A, 0, -va);
    spfa( 0 );
    recover(c, d);
    return flag;
}

bool check2(int c, int d, int vd, int va){//>
    init(c, d);
    int t = va - vd - 1;
    adde(B, c, t);
    adde(0, d, vd); adde(d, 0, -vd);
    adde(0, A, va); adde(A, 0, -va);
    spfa( 0 );
    recover(c, d);
    return flag;
}

bool check3(int c, int d, int vd, int va){//=
    init(c, d);
    int t = vd - va;
    adde(c, B, t); adde(B, c, -t);
    adde(0, d, vd); adde(d, 0, -vd);
    adde(0, A, va); adde(A, 0, -va);
    spfa( 0 );
    recover(c, d);
    return flag;
}

int main(){
    scanf("%d%d%d", &n, &A, &B);
    for(int i=1; i<=n; i++) {
        char s[100]; scanf("%s", s+1);
        for(int j=1; j<=n; j++){
            if(s[j]=='+') adde(i, j, -1);
            if(s[j]=='-') adde(j, i, -1);
            if(s[j]=='=') adde(j, i, 0), adde(i, j, 0);
        }
    }
    for(int i=1; i<=n; i++) adde(0, i, 3), adde(i, 0, -1);//限定所有元素在1~3 
    for(int i=1; i<n; i++)
        for(int j=i+1; j<=n; j++) {//枚举右边两个 
            if(i==A || j==B || i==B || j==A) continue;
            int v1 = 0, v2 = 0, v3 = 0;
            //以A+B<C+D为例，因为差分约束必须是差的形式，所以我们将式子变形 
            //B-C<D-A然后枚举D,A的取值，就得到了一个关于两个数差的不等式。
            for(int d=1; d<=3; d++){
                for(int a=1; a<=3; a++) {
                    int t1 = check1(i, j, d, a); if( !t1 ) v1++;
                    int t2 = check2(i, j, d, a); if( !t2 ) v2++;
                    int t3 = check3(i, j, d, a); if( !t3 ) v3++;
                }
            }
            if(v1 && !v2 && !v3) ans[1]++;
            if(!v1 && v2 && !v3) ans[2]++;//因为只有结果保证惟一的选法才统计在内 
            if(!v1 && !v2 && v3) ans[3]++;//如果枚举的C,D的值不同，得到的与A+B的关系不同的话，那么方案不合法。
        }
    printf("%d %d %d\n", ans[2], ans[3], ans[1]);
    return 0;
}