题目描述
LuxakyLuee是一个非常爱玩的人。
有一天一个叫Kevin·Feng的人向LuxakyLuee发来游戏邀请。
Kevin·Feng摆了一地的键盘冒,上面都是英文字母,让LuxakyLuee从其中任选k个。
选完之后进入清算时间,每一个键帽的分数是总共选取的的该种键帽字母的数量。
如果LuxakyLuee足够机智他能得多少分呢。
输入
多组测试数据。
每组数据有两行。
第一行为n和k,n为键盘冒的数量,k为选取的键盘冒的个数。(1?≤?k?≤?n?≤?100000).
第二行为地上的键盘冒(均为大写字母)。
输出
每组数据输出一行,为LuxakyLuee的最高可能得分。
样例输入
15 10
DZFDFZDFDDDDDDF
6 4
YJSNPI
样例输出
82
4
解析:哪个键帽个数最多,我们先优先取哪个键帽,因为记分规则是,取过来每个相同的键帽的个数,例如取过来DDDDF,记分就是4*4*4*4+1*1.其实就是各个键帽个数的平方和。所以我们从大到小排序,然后优先选择多的键帽字母,直到选择次数用完。
#include <bits/stdc++.h>
#include <string.h>
#include <algorithm>
using namespace std;
long long a[30]; //记录各个键帽的数量
char su[100005]; //输入字符串存入su
bool cmp(long long a,long long b){return a>b; //sort的排序方式,从大到小
}
int main()
{long long n,k,len,i,sum;while(~scanf("%lld%lld",&n,&k)){sum=0; //得分 for(i=0;i<30;i++) a[i]=0; //初始化0 memset(su,'\0',sizeof(su)); //初始化 scanf("%s",su);len=strlen(su);for(i=0;i<len;i++) a[su[i]-'A'+1]++; //对应字母个数计数 sort(a+1,a+27,cmp); //从大到小排序 for(i=1;;i++){if(k>=a[i]){ //如果抓取次数大于字母个数 sum=sum+a[i]*a[i]; //累加得分 k=k-a[i]; //次数消耗 }else{sum=sum+k*k;k=0;}if(k==0) break; //次数用完,退出 }printf("%lld\n",sum);}return 0;
}