N个金币(编号为1-N)中有一枚重量不同的假金币(真金币重量相同),利用唯一的一台天平将金币分组称量可以找出假金币。
第一行是一个整数M表示一共有多少组测试数据,每组测试数据的第一行输入2个空格隔开的整数N和K,N是金币的总数(2~1000),K是称重的次数(1~100)。随后2K行记录称量的情况和结果,连续2行记录一次称量:第1行首先是Pi(1-N/2),表示两边托盘放置的金币数目,随后是左边托盘中Pi个金币编号和右边托盘中Pi个金币编号,所有数之间都由空格隔开;第2行用<、>和=记录称量结果。
对每组测试数据输出假金币的编号,如果称量记录无法确定假金币,输出0。
1 5 3 2 1 2 3 4 < 1 1 4 = 1 2 5 =
3
对于这个问题,首先我们想到的是分治求解,但这里我想的是一种比较巧妙的方法,我们利用对其中的真假金币进行模糊判断,判断依据是两边金币是否相等,因为只有一个假金币,所以如果两边相等,就说明都是真的金币,就在一个标志数组的对应位置值加一;不等则减一,最后如果有一个唯一最小值,则那个就是假金币,找不到则不能确定。
代码:
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
int M;
cin>>M;
while(M--){
int N,K;
cin>>N>>K;
int arr1[N],arr2[N],arr3[N];
for(int i = 0 ; i < N ; i++)
{
arr1[i] = i+1;
arr2[i] = 1;
}
for(int i = 0 ; i < K ; i++)
{
int Pi;
cin>>Pi;
int right[Pi],left[Pi];
for(int i = 0 ; i < Pi ; i++)
{
cin>>left[i];
}
for(int i = 0 ; i < Pi ; i++)
{
cin>>right[i];
}
char ch;
cin>>ch;
if(ch == '<'||ch == '>')
{
for(int i = 0 ; i < Pi ; i++)
{
for(int j = 0 ; j < N ; j++)
{
if(arr1[j] == left[i]) arr2[j]--;
if(arr1[j] == right[i]) arr2[j]--;
}
}
}
if(ch == '=')
{
for(int i = 0 ; i < Pi ; i++)
{
for(int j = 0 ; j < N ; j++)
{
if(arr1[j] == left[i]) arr2[j]++;
if(arr1[j] == right[i]) arr2[j]++;
}
}
}
}
for(int i = 0; i < N; i++)
{
arr3[i] = arr2[i];
}
sort(arr2,arr2+N);
if(arr2[0] == arr2[1]) cout<<0<<endl;
if(arr2[0] < arr2[1])
{
int flag = arr2[0];
for(int i = 0 ; i < N ; i++)
{
if(flag == arr3[i])
cout<<i+1<<endl;
}
}
}
return 0;
}