笔试编程题 字符串价值 Java题解 有一种有趣的字符串价值计算方式:统计字符串中每种字符出现的次数,然后求所有字符次数的平方和作为字符串的价值 例如: 字符串“abacaba“,里面包括4个‘a‘

题目描述

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

输出描述:

输出一个整数,表示得到的最小价值

输入

aba 1

输出

2

Java题解

这道题做了快半个小时，最后数组掐尖去除改了N次终于对了。

大致思路就是先把每一个都有多少算出来，最后把数字大的先变小，因为是平方关系，只要把大数该小就能得到相对小的数

比如：a - 1   b - 6  c - 2  需要去除2个字符，肯定是去掉两个b

原来是1*1+2*2+6*6，现在是1*1+2*2+4*4，现在这个就是最小的答案

import java.util.*;
public class Main {
    public static void main(String[] args){
        //输入
        Scanner sc = new Scanner(System.in);
        String str = sc.nextLine();
        int max = sc.nextInt();
        if(str.length() < 1){
            System.out.println(0);
        }
        //计数
        HashMap<Byte, Integer> map = new HashMap<Byte, Integer>();
        byte[] bytes = str.getBytes();
        for(int i = 0; i < bytes.length; i++){
            if(map.containsKey(bytes[i])){
                map.put(bytes[i], map.get(bytes[i]) + 1);
            }else{
                map.put(bytes[i], 1);
            }
        }
        //System.out.println(map);
        int size = map.size();
        int[] score = new int[size];
        int i = 0;
        Set<Byte> bytesKey = map.keySet();
        for (Byte aByte : bytesKey) {
            score[i] = map.get(aByte);
            i++;
        }
        //重拍
        if(max != 0) {
            Arrays.sort(score);
            //System.out.println(Arrays.toString(score));
            //剔除
            while (max > 0) {
                int scoreSize = score.length - 1;
                int last = score[scoreSize];
                for (i = scoreSize; i >= 0; i++) {
                    if (scoreSize >= 0 && score[scoreSize] == last && max > 0) {
                        score[scoreSize]--;
                        scoreSize--;
                        max--;
                    } else {
                        break;
                    }
                }
            }
        }
        //重算
        int sum = 0;
        for (int scoreOnce : score) {
            sum = sum + (scoreOnce * scoreOnce);
        }
        System.out.println(sum);
    }
}