【Lintcode】1893. The Valid String

题目地址：

https://www.lintcode.com/problem/the-valid-string/description

给定一个字符串，题目保证只有英文字母。如果其每个字母出现的次数相同，则返回YES，否则如果删去其中一个字符，能使得每个字母出现次数相同，则也返回YES，否则返回NO。

由上面可以知道，返回YES的情况有下面几个：
1、每个字母出现次数一样；
2、所有的字母只有两种出现次数，并且或者出现次数多的字符只有一个且它出现的次数减去$1$就能和别人一样，或者出现次数少的字符只有一个且它只出现了$1$次。

所以思路就是，先统计对于每个次数，有几个字符出现了这么多次，$0$次不参与统计。统计完之后，如果发现次数做成的集合是单元集，说明每个字符出现次数一样，返回YES；如果次数做成的集合的元素个数大于$2$，那么只删一个字符是无法让所有字符出现次数一样的，返回NO；接下来考虑次数做成的集合的元素个数等于$2$的情况，如果出现次数多的字符个数等于$1$并且删掉它就能达到和别的字符出现次数一样，或者出现次数少的字符个数等于$1$并且它只出现了一次，那都返回YES，其余情况都返回NO。也就是删除的时候，只需要考虑两种情况，是删出现次数多的，还是少的。这样考虑问题会比较清楚。代码如下：

import java.util.HashMap;
import java.util.Map;

public class Solution {
    /**
     * @param s: a String
     * @return: if valid return "YES" else return "NO"
     */
    public String isValid(String s) {
        // write your code here
        int[] count = new int[26];
        for (int i = 0; i < s.length(); i++) {
            count[s.charAt(i) - 'a']++;
        }
        
        // 出现了key次的字符有value个
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < count.length; i++) {
            if (count[i] == 0) {
                continue;
            }
            
            map.put(count[i], map.getOrDefault(count[i], 0) + 1);
        }
        
        if (map.size() == 1) {
            return "YES";
        }
        if (map.size() > 2) {
            return "NO";
        }
        
        // cnt存字符出现的次数
        int[] cnt = new int[2];
        int idx = 0;
        for (int c : map.keySet()) {
            cnt[idx++] = c;
        }
        
        // idx表示次数多的那个对应的下标
        idx = cnt[0] > cnt[1] ? 0 : 1;
        
        // 如果删掉出现次数多的那个字符能达到效果，则返回YES
        if (map.get(cnt[idx]) == 1 && cnt[idx] - cnt[idx ^ 1] == 1) {
            return "YES";
        }
        // 如果删掉出现次数少的那个字符能达到效果，则也返回YES
        if (map.get(cnt[idx ^ 1]) == 1 && cnt[idx ^ 1] == 1) {
            return "YES";
        }
        
        // 其余情况返回NO
        return "NO";
    }
}

时间复杂度$O(n)$，空间$O(26)=O(1)$。