Description
In order to prevent the worst-case scenario, Akira should distinguish the devilish from the divine. But how? They looked exactly alike and he could not distinguish one from the other solely by their appearances. He still had his last hope, however. The members of the divine tribe are truth-tellers, that is, they always tell the truth and those of the devilish tribe are liars, that is, they always tell a lie.
He asked some of them whether or not some are divine. They knew one another very much and always responded to him "faithfully" according to their individual natures (i.e., they always tell the truth or always a lie). He did not dare to ask any other forms of questions, since the legend says that a devilish member would curse a person forever when he did not like the question. He had another piece of useful informationf the legend tells the populations of both tribes. These numbers in the legend are trustworthy since everyone living on this island is immortal and none have ever been born at least these millennia.
You are a good computer programmer and so requested to help Akira by writing a program that classifies the inhabitants according to their answers to his inquiries.
Input
n p1 p2
xl yl a1
x2 y2 a2
...
xi yi ai
...
xn yn an
The first line has three non-negative integers n, p1, and p2. n is the number of questions Akira asked. pl and p2 are the populations of the divine and devilish tribes, respectively, in the legend. Each of the following n lines has two integers xi, yi and one word ai. xi and yi are the identification numbers of inhabitants, each of which is between 1 and p1 + p2, inclusive. ai is either yes, if the inhabitant xi said that the inhabitant yi was a member of the divine tribe, or no, otherwise. Note that xi and yi can be the same number since "are you a member of the divine tribe?" is a valid question. Note also that two lines may have the same x's and y's since Akira was very upset and might have asked the same question to the same one more than once.
You may assume that n is less than 1000 and that p1 and p2 are less than 300. A line with three zeros, i.e., 0 0 0, represents the end of the input. You can assume that each data set is consistent and no contradictory answers are included.
Output
Sample Input
2 1 1
1 2 no
2 1 no
3 2 1
1 1 yes
2 2 yes
3 3 yes
2 2 1
1 2 yes
2 3 no
5 4 3
1 2 yes
1 3 no
4 5 yes
5 6 yes
6 7 no
0 0 0
Sample Output
no
no
1
2
end
3
4
5
6
end
Source
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#include <utility>
#include <iomanip>
#include <string>
#include <cmath>
#include <queue>
#include <map>
#define LOCAL
const int MAXN = + ;
const int MAX = + ;
using namespace std;
struct DATA{
int a, b;
}data[MAXN];
int n, p1, p2, q;
int parent[MAXN] ,val[MAXN];
int num[MAXN];//num[i]代表如果i是并查集中的根，那么他在AB表中的位置就num[i]
int f[MAXN][MAXN], cnt;
int pre[MAXN][MAXN];//DP中记录状态转移过来的位置
int Ans[MAXN];//表示答案中被选为好人或者不是好人 int find(int x){
int f = x, tmp = ;
while (x != parent[x]){
tmp += val[x];
x = parent[x];
}
parent[f] = x;
val[f] = tmp % ;
return x;
} void init(){
n = p1 + p2;//即总的人数
//val为0代表相同集合，1代表不同的集合
for (int i = ; i <= n; i++){
val[i] = ;
parent[i] = i;
}
for (int i = ; i <= q; i++){
int x, y;
char str[];
scanf("%d%d", &x, &y);
scanf("%s", str);
if (find(x) == find(y)) continue;//在同一个集合之内就不用考虑了
else{
if (str[] == 'y'){ //一定是一种集合之内
int xx = find(x), yy = find(y);
parent[xx] = y;
val[xx] = (-val[x] + ) % ;
}else{
int xx = find(x), yy = find(y);
parent[xx] = y;
val[xx] = (-val[x] + ) % ;
}
}
}
}
//计算出各个并查集的同类和不同类
void prepare(){
memset(data, , sizeof(data));
cnt = ;
for (int i = ; i <= n; i++){
if (parent[i] != i) continue;
num[i] = ++cnt;
}
for (int i = ; i <= n; i++){
int xx = find(i);
if (val[i] == ) data[num[xx]].a++;//和它同一类
else data[num[xx]].b++;//不同类
}
}
void dp(){
memset(f, , sizeof(f));
//f[i][j]表示到了第i个好人有j个的时候的方案数量,注意只要保存3个量就可以了
f[][] = ;
for (int i = ; i <= cnt; i++)
for (int j = p1; j >=; j--){
if (j - data[].a >= ) f[i][j] += f[i - ][j - data[i].a];
if (j - data[].b >= ) f[i][j] += f[i - ][j - data[i].b];
if (f[i][j] >= ) f[i][j] = ;//不要做太大了
if (f[i][j] == ){//有解
if (j - data[i].a >= && f[i - ][j - data[i].a] == ) pre[i][j] = ;//a类选为好人
if (j - data[i].b >= && f[i - ][j - data[i].b] == ) pre[i][j] = ;//b类选为好人
}
}
if (f[cnt][p1] != ){printf("no\n");return;}
int last = p1;
for (int i = cnt; i >= ; i--){
Ans[i] = pre[i][last];
if (Ans[i] == ) last -= data[i].a;
else last -= data[i].b;
}
for (int i = ; i <= n; i++){
int xx = find(i);
if (Ans[num[xx]] == && val[i] == ) printf("%d\n", i);
if (Ans[num[xx]] == && val[i] == ) printf("%d\n", i);
}
printf("end\n");
} int main(){
int T;
#ifdef LOCAL
freopen("data.txt", "r", stdin);
freopen("out.txt", "w", stdout);
#endif
while (scanf("%d%d%d", &q, &p1, &p2)){
if (q == && p1 == && p2 == ) break;
init();
prepare();
dp();
//printf("%d", data[1].b);
}
return ;
}