题目:
小b有n张牌。
现在她想把牌分组,使得每组都是长度为W的顺子,即由连续W个数组成。
请问小b能做到吗?
输入
第一行输入一个数n,表示手牌张数;
第二行输入n个非负整数,表示每张牌的数字,以空格隔开;
第三行输入一个数,表示每组大小W;
其中1≤W≤n≤10000,任意牌的数字hand[i]满足0≤hand[i]≤10^9
输出
可以分组,输出“true”;
不能分组,输出“false”。
输入样例
输出样例
解法
用map记录出现的数,每次取剩下的最小的数,然后删除相应的连续的数,当出现为负数的时候,则为false
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56
| #include<stdio.h> #include<algorithm> #include<queue> #include<map> using namespace std; map<int,int> m; int main() { int n; scanf("%d", &n); int MAX = 0; for(int i = 0; i < n; i++){ int a; scanf("%d", &a); MAX = max(a,MAX); m[a]++; } int q; scanf("%d", &q); if(n%q != 0){ printf("false"); return 0; } map<int,int> ::iterator it; while(n){ /*for(it = m.begin(); it != m.end(); it++){ printf("%d %d\n", it->first, it->second); } printf("\n\n");*/ int p_1 = m.begin()->first; int p_2 = m.begin()->second; if(p_1 + q - 1 > MAX){ printf("false"); return 0; } for(int i = 1; i < q; i++){ if(m.find(i+p_1) == m.end()){ printf("false"); return 0; } m[i+p_1] -= p_2; if(m[i+p_1] < 0){ printf("false"); return 0; } if(m[i + p_1] == 0){ //pair<int,int> a(i+p_1,0); m.erase(i+p_1); } } m.erase(p_1); n -= p_2 * q; } printf("true"); return 0; }
|