Wrong answer on test 6:
http://codeforces.me/group/2TSikdYNgC/contest/204801/problem/F
#include <iostream>
#include "stdio.h"
#include <algorithm>
#include <cmath>
using namespace std;
bool cmp (int i, int j){
return i > j;
}
int main(int argc, const char * argv[]) {
int n, k;
cin >> n >> k;
char arr[n];
for (int i = 0; i < n; i++)
{
cin >> arr[i];
}
sort(arr,arr + n);
int summ[n];
int s = 0;
summ[0] = 0;
for (int i = 0; i < n; i++)
{
if (arr[i] == arr[i+1]) {
summ[s]++;
} else
{
s++;
summ[s]=0;
}
}
int res = 0;
sort(summ,summ + s,cmp);
int b[s];
int l = 0;
for (int i = 0; i < s; i++) {
summ[i]++;
if (k <= summ[i])
{
b[l++] = k;
k = 0;
} else {
b[l++] = summ[i];
k -= summ[i];
}
}
for (int i = 0; i < l; i++)
{
res += b[i]*b[i];
}
cout << res;
}
I'm using XCode c++ compiler.
And test 6 (2 2 BL) works great.
Why wrong answer on codeforces? Thank you.



