爱奇艺 字符串价值

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

输出例子1:

2

 思路：平方和可以拆开来看，相同字符时，第一个字母就为1，第二个就加3，第三个就加5，删除时，就倒着减。

import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String s = sc.nextLine();
        int[] a = new int[s.length()];
        int sum = 0;
        int index = 0;
        int k = sc.nextInt();
        HashMap<Character,Integer> map = new HashMap<>();
        for(int i = 0;i<s.length();i++){
            char c = s.charAt(i);
            if(map.containsKey(c)){
                int v = map.get(c)+1;
                map.put(c,v);
                a[index] = 2*v - 1;
                sum+=a[index];
                index++;
            }else{
                map.put(c,1);
                a[index++] = 1;
                sum+=1;
            }
        }
        Arrays.sort(a);
        index = s.length() - 1;
        while(k>0){
            k--;
            sum -= a[index--];
        }
        System.out.println(sum);
    }
}