爱奇艺 字符串价值 贪心

题目描述

有一种有趣的字符串价值计算方式:统计字符串中每种字符出现的次数,然后求所有字符次数的平方和作为字符串的价值
例如: 字符串"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

输入

输出

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn = 26;
int arr[maxn];
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
    string s;
    int n;
    while(cin>>s>>n)
    {
        int len = s.length();
        memset(arr,0,sizeof(arr));
        for(int i=0;i<len;++i)
        {
            arr[s[i]-'a']++;
        }
        //sort(arr,arr+26,cmp);
        /*
        for(int i=0;i<26;++i)
        {
            cout<<arr[i]<<" ";
        }
        cout<<endl;
        */
        for(int i=0;i<n;++i)
        {
            sort(arr,arr+26,cmp);
            arr[0]--;
        }
        /*
        for(int i=0;i<26;++i)
        {
            cout<<arr[i]<<" ";
        }
        cout<<endl;
        */
        int ans=0;
        for(int i=0;i<26;++i)
        {
            ans+=arr[i]*arr[i];
        }
        cout<<ans<<endl;
    }
    return 0;
}