牛客真题（22）-字符串价值

今天继续刷牛客真题，字符串价值

分析：
对于一个给定的字符串，第一步，需要统计字符串中每一个字符的个数（通过数组实现，也可以通过map映射实现，但是这个题目数组实现比较方便），第二步，对个数值数组进行排序，并且每次对最大的数进行减一，直到k为0，第三步，将数组每个元素求平方后求和。

问题：
1、Python中，对字符串的个数统计函数Counter（）；
2、Python中，对每个元素进行求平方，用map映射。
3、C++中数组排序，sort（a,a+n)；

附上C++代码：

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<string>
using namespace std;
int main()
{
    string s;
    getline(cin,s);
    int k;
    cin>>k;
    int a[26]={0};
    for(int i=0;i<s.length();i++)
    {
        a[s[i]-'a']++;
    }
    while(k)
    {
        sort(a,a+26);
        a[25]--;
        k--;
    }
    int sum=0;
    for(int i=0;i<26;i++)
    {
        sum+=a[i]*a[i];
    }
    cout<<sum<<endl;
    return 0;
}

附上Python代码：

from collections import Counter

s=input()
k=int(input())
l=list(Counter(s).values())
while k>0:
    l.sort()
    l[-1]-=1
    k-=1
print(sum(map(lambda x:x**2,l)))