题目地址:
https://www.lintcode.com/problem/the-valid-string/description
给定一个字符串,题目保证只有英文字母。如果其每个字母出现的次数相同,则返回YES,否则如果删去其中一个字符,能使得每个字母出现次数相同,则也返回YES,否则返回NO。
由上面可以知道,返回YES的情况有下面几个:
1、每个字母出现次数一样;
2、所有的字母只有两种出现次数,并且或者出现次数多的字符只有一个且它出现的次数减去
1
1
1就能和别人一样,或者出现次数少的字符只有一个且它只出现了
1
1
1次。
所以思路就是,先统计对于每个次数,有几个字符出现了这么多次, 0 0 0次不参与统计。统计完之后,如果发现次数做成的集合是单元集,说明每个字符出现次数一样,返回YES;如果次数做成的集合的元素个数大于 2 2 2,那么只删一个字符是无法让所有字符出现次数一样的,返回NO;接下来考虑次数做成的集合的元素个数等于 2 2 2的情况,如果出现次数多的字符个数等于 1 1 1并且删掉它就能达到和别的字符出现次数一样,或者出现次数少的字符个数等于 1 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(n) O(n),空间 O ( 26 ) = O ( 1 ) O(26)=O(1) O(26)=O(1)。