题目描述

有一种有趣的字符串价值计算方式:统计字符串中每种字符出现的次数,然后求所有字符次数的平方和作为字符串的价值

例如: 字符串"abacaba",里面包括4个'a',2个'b',1个'c',于是这个字符串的价值为4 * 4 + 2 * 2 + 1 * 1 = 21

牛牛有一个字符串s,并且允许你从s中移除最多k个字符,你的目标是让得到的字符串的价值最小。

输入描述:

输入包括两行,第一行一个字符串s,字符串s的长度length(1 ≤ length ≤ 50),其中只包含小写字母('a'-'z')。 第二行包含一个整数k(0 ≤ k ≤ length),即允许移除的字符个数。

输出描述:

输出一个整数,表示得到的最小价值

示例1

输入

aba 1

输出

2

n^2是曲线形增长。

要使价值最小，肯定要把每个的n值尽可能的降低，而不是单独只降低一个。

比如 2 * 2 + 2 * 2，只能降低2次

1 * 1 + 1 * 1 <= 2 * 2 + 0 * 0。

每次只降低最大的，用set存数据会好点，因为会频繁排序。

不过数组只有常数26长，重复排序的性能耗费不大。

#include<iostream>
#include<algorithm>
#include<string>
using namespace std;

int main()
{
    string str;
    int n;
    cin >> str;
    cin >> n;
    int alpa[26] = {0};
    for(int i = 0; i < str.size(); i++)
    {
        ++alpa[str[i] - 'a'];
    }
    for(int i = 0; i < n; i++)
    {
        sort(alpa, alpa + 26);
        alpa[25]--;
    }
    int ans = 0;
    for(int i = 0; i < 26; i++)
    {
        ans += alpa[i] * alpa[i];
    }
    cout << ans << endl;
    return 0;
}