字符串价值

题目描述

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

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

bool cmp(const int &x,const int &y)
{
    return x>y;
}
int main()
{
   string str;
   int de_num=0,sum=0;
   int dele_num,arr_size=0;
   int arr[26];fill(arr,arr+26,0);//数组过大时间会难以通过 
   getline(cin,str);
   cin>>dele_num;
   for(int i=0;i<str.size();i++)
   arr[str[i]-'a']++;//注意是 [str[i]-'a']
   sort(arr,arr+26,cmp);//第一遍排好序（降序） 
   for(int i=0;i<26;i++)
   if(arr[i]!=0)arr_size++;

    int k=0;
    while(1)
    {
        if(de_num==dele_num)break;//要考虑dele_num==0的情况 
        arr[k]--;de_num++;
        sort(arr+0,arr+de_num,cmp);//继续排序 
    } 

    for(int i=0;i<arr_size;i++)
    sum+=arr[i]*arr[i];    
    cout<<sum;

    return 0;
}
/*

用例:
wersrsresesrsesrawsdsw
11
对应输出应该为:
23
你的输出为:
31*/