【水题】CodeForce 1183D Candy Box (easy version)

这段时间要沉迷刷题一段时间了，就让CSDN陪我一起吧！

一、题目大意

题目的意思就是一共有q个查询，在每个查询中第一行是一个整数n，表示糖果的数量，第二行为n个整数，表示每个糖果的种类，要求凑礼品盒，求得可以凑礼品盒的最大尺寸(最大糖果的数量)。凑的原则是不同的种类所有的数量不能相同。

二、题目思路以及AC代码

本来是道水题，可能有一段时间没做了，也花了点时间。

主要的思路其实就是，首先在输入的时候，对不同种类的糖果计数，然后按照数量从大到小进行排序，然后开始求结果，直接扫描这个排好序的数组，根据题目的要求选择即可，就是一个简单的实现，不明白的可以参照我的代码。

但我在第一次实现的时候出现了TLE，当然很快就找到了原因，出在了初始化上，我直接定义了一个长度为2*10⁵的数组，然后每次查询的时候对其初始化，这样的复杂度就成了O(n²)了，自然会超时，然后我又看了一下题目，发现a_i的范围在1-n之间，这样就可以每次只初始化到n，这样就是O(n)的复杂度了，至于为什么，和题目给的输入限制有关，这里不过多解释。

下面给出AC代码：

#include <iostream>
#include <algorithm>
#define MAXN 200010
using namespace std;

int q;
int n;
int a[MAXN];
int arr[MAXN];

void init(){
    for (int i=0;i<MAXN;i++){
        a[i] = 0;
        arr[i] = i;
    }
}

bool cmp(int i, int j){
    return a[i] > a[j];
}

int main()
{
    cin >> q;
    while(q--){
        cin >> n;
        for (int i=1;i<=n;i++){
            a[i] = 0;
            arr[i] = i;
        }

        int tmp;
        int tsize = 0;
        for (int i=1;i<=n;i++){
            cin >> tmp;
            if (!a[tmp]) tsize++;
            a[tmp]++;
        }

        sort(arr+1, arr+n+1, cmp);

        int cnt = 0;
        int last = a[arr[1]];
        int num = last;
        cnt += last--;
        for (int i=2;i<=tsize;i++){
            if (!a[arr[i]]) continue;
            if (a[arr[i]] != num){
                if (a[arr[i]] >= last){
                    cnt += last;
                }
                else {
                    last = a[arr[i]];
                    cnt += last;
                }
                last--;
                num = a[arr[i]];
            }
            else {
                cnt += last--;
            }
            if (last <= 0) break;
        }

        cout << cnt << endl;
    }

    return 0;
}

如果有问题，欢迎大家指正！！！