爱奇艺校招map优先队列笔试题字符串的价值

题目描述

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

 

这里先用map将字符和出现频率map到一起，最后删除的时候需要从频率最高到低动态删除，所以这里需要使用一个优先队列。

注意这里的实现方法

#include <iostream>
#include <string>
#include <queue>
#include <map>
using namespace std;

int  main()
{
    string str;
    int k;
    cin>>str;
    cin>>k;
    map<char,int> mp;
    for(auto c:str){
        mp[c]++;
    }
    priority_queue<pair<int,char>> pq;
    for(auto it=mp.begin();it!=mp.end();++it){
        pq.push(make_pair(it->second, it->first));
    }
    for(int i=0;i<k;i++){
        int value = pq.top().first;
        int key = pq.top().second;
        pq.pop();
        pq.push(make_pair(value-1, key));
    }
    
    int score = 0;
    while(!pq.empty()){
        int value = pq.top().first;
        score += value * value;
        pq.pop();
    }
    cout<<score<<endl;
    

}