爱奇艺笔试题之字符串价值

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

思路：统计每个字符出现的次数，然后将出现最多的次数-1，直至减到k次，最后计算得到结果。

import java.util.Arrays;
import java.util.Scanner;

public class Aiqiyi04 {
    public static void main(String[] args){
//        Scanner in = new Scanner(System.in);
//        String line = in.nextLine();
        String line = "oohlaefleecoookjkoippolm";
//        int k = Integer.parseInt(in.nextLine());
        int k = 13;
        int[] times = new int[26];
        char[] array = line.toCharArray();

        for (int i=0; i<array.length; i++){
            times[array[i]-'a'] += 1;
        }

        Arrays.sort(times);
        for (int i=0; i<k; i++){
            Arrays.sort(times);
            times[25] --;
        }
        Arrays.sort(times);

        int sum=0;
        for (int i=times.length-1; i>=0; i--){
            if (times[i] > 0){
                sum += times[i]*times[i];
            } else {
                break;
            }
        }

        System.out.println(sum);
    }
}