LeetCode-1641. 统计字典序元音字符串的数目
题目描述:
给你一个整数 n,请返回长度为 n 、仅由元音 (a, e, i, o, u) 组成且按 字典序排列 的字符串数量。
字符串 s 按 字典序排列 需要满足:对于所有有效的 i,s[i] 在字母表中的位置总是与 s[i+1] 相同或在 s[i+1] 之前。
上来先思考了下dp和二叉树,感觉都不够直接,后来按照第一位分情况讨论,问题就可以转化成组合数学的问题。
思路:当字符串以a为开头的时候,由这样几种情况
- 字符串只有a一种字母,此时排序只有一种可能 result += 1
- 字符串除a以外有1种字母,此时这另外1种字母有C(1,4)种可能(e,i,o,u),其位置有C(1, n-1)种可能。
- 字符串除a以外有2种字母,此时这另外2种字母有C(2,4)种可能(e,i,o,u),其位置有C(2, n-1)种可能。
- 字符串除a以外有3种字母,此时这另外3种字母有C(3,4)种可能(e,i,o,u),其位置有C(3, n-1)种可能。
- 字符串除a以外有4种字母,此时这另外4种字母有C(4,4)种可能(e,i,o,u),其位置有C(4, n-1)种可能。
其余字母开头情况类似,代码实现如下:
class Solution {
public:
int countVowelStrings(int n) {
--n;
if(n == 0){return 5;}
int result = 0;
result +=1 + c(4, 1)*c(n, 1) + c(4, 2) * c(n, 2) + c(4, 3) * c(n, 3) + c(4, 4) * c(n, 4);
result +=1 + c(3, 1)*c(n, 1) + c(3, 2) * c(n, 2) + c(3, 3) * c(n, 3);
result +=1 + c(2, 1)*c(n, 1) + c(2, 2) * c(n, 2);
result +=1 + c(1, 1)*c(n, 1);
result += 1;
return result;
}
//从n中选出m
int c(int n, int m){
if(m > n) return 0;
long long a = 1;
long long b = 1;
while(m){
a *= n;
b *= m;
--n;
--m;
}
return a/b;
}
};