ZCMU--1484: 签到题(C语言)

题目描述

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;
}