Description
有一群喜欢带小红帽的家伙举行了一场别开生面的聚会,然而某些人被邪恶的WM讲帽子上涂了大灰狼的图标,可是每个人看不到自己头上的帽子有没有被涂,他们只能看到其他人头上的帽子是不是大灰狼的图案,现在告诉你每个人看到的别人头上大灰狼帽子的数量,聪明的你啊,能不能判断出来一共有多少个人头上被汪淼涂了可恶的大灰狼呢,当然如果你发现有些人撒谎的话,就直接输出-1吧
Input
包含多组测试数据,每组数组有两行
第一行读入一个n,代表聚会的人数 (n<=100)
第二行一次读入n个数,a[i]代表第i个人看到的其他人头上的大灰狼的个数
第一行读入一个n,代表聚会的人数 (n<=100)
第二行一次读入n个数,a[i]代表第i个人看到的其他人头上的大灰狼的个数
Output
每组数据输出一个数,多少人被涂了大灰狼
Sample Input
3
1 2 1
4
1 1 1 2
1 2 1
4
1 1 1 2
Sample Output
2
-1
-1
算法:
起先看到这个题目时,不知道如何下手,但迅速找一种情况来分析下,比如有5个家伙
若都没有大灰狼图标则:0 0 0 0 0
如果有一个: 0 1 1 1 1
如果有两个:1 1 2 2 2
如果有三个:2 2 2 3 3
如果有四个:3 3 3 3 4
如果有5个:4 4 4 4 4
可以看出上面的规律,若有n个人,其中m个有大灰狼图标的话,其中有结果有m个m-1和n-m个m(如果m为0的话,是n个0)
其他都是细节问题,下面上代码
#include<iostream>
#include<vector>
using namespace std;#define num 100int main()
{int n;while(cin>>n){bool flag=true;int a[100]={0};vector<int>vetmp;for(int i=0;i<n;i++){int input;cin>>input;//如果输入大于人数减一if(input>n-1&&flag==true){flag=false;cout<<-1<<endl;}elsea[input]++;//用于统计多少人数,相当于计数排序}if(flag==false)continue;int sum=0,max=-1,min=-1;for(int i=0;i<num;i++){if(a[i]!=0){sum++;if(min==-1)min=i;else max=i;}}//end for//这种情况是都没狼或者都有狼if(sum==1&&max==-1&&(min==0||min==n-1)){if(min==0)cout<<0<<endl;else cout<<n<<endl;}//有一些人有狼,有一些人没狼else if(sum==2&&a[min]==max&&min==max-1){cout<<min+1<<endl;}else{cout<<-1<<endl;}}//end while
}