题目描述

西行寺幽幽子的庭院中有n棵樱花树，每棵樱花树都有一个美丽值a_i。

幽幽子决定将这些树分成不超过m组，每组树的美丽度为组内所有美丽度不同的树的美丽度之和（比如1,1,1,2,2,2这组的美丽度为1+2=3。

求最优分组方案下，所有组的美丽度总和。

输入

第一行输入两个整数n, m, 表示樱花树数量和最大分组数量。

接下来一行输入n个整数，表示每棵树的美丽度a_i。

输出

输出仅一个整数，表示最大的美丽度总和。

样例

输入数据 1

12 2
1 2 3 4 2 3 4 5 3 4 5 6

输出数据 1

35

数据规模

对于前 30% 的数据：1 \le n,m,a_i \le 100

对于前 50% 的数据：1 \le n \le 10^4

对于前 70% 的数据：1 \le a_i \le 10^6

对于另外 10% 的数据：m=n

对于 100% 的数据：1 \le n,m \le 10^6,1 \le a_i \le 10^9

样例解释

最优分组方案之一为1,2,3,4,3,5和2,4,3,4,5,6，美丽度总和为(1+2+3+4+5)+(2+3+4+5+6)=35

MY CODE

#include <bits/stdc++.h>

using namespace std;

int main() {
    int n, m;
    scanf("%d%d", &n, &m);
    vector<int> beauty(n);
    for (int i = 0; i < n; i++) {
        scanf("%d", &beauty[i]);
    }
    long long tot = 0;
    if (m > 1) {
        for (int b : beauty) {
            tot += b;
        }
        printf("%lld\n", tot);
    } else {
        for (int b : beauty) {
            tot += b;
        }
        printf("%lld\n", tot);
    }

    return 0;
}

请大佬们帮忙调试一下，连结果都不对，提供AC代码者必关！！！ 在线等，很急！！！