题目:

小b有n张牌。

现在她想把牌分组,使得每组都是长度为W的顺子,即由连续W个数组成。

请问小b能做到吗?

输入

第一行输入一个数n,表示手牌张数; 第二行输入n个非负整数,表示每张牌的数字,以空格隔开; 第三行输入一个数,表示每组大小W; 其中1≤W≤n≤10000,任意牌的数字hand[i]满足0≤hand[i]≤10^9

输出

可以分组,输出“true”; 不能分组,输出“false”。

输入样例

9
1 2 3 6 2 3 4 7 8
3

输出样例

true

解法

用map记录出现的数,每次取剩下的最小的数,然后删除相应的连续的数,当出现为负数的时候,则为false

代码

#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;
}