你有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;
}