题目太长放链接
力扣5974
本来还以为需要高精度,大概算了下long long 勉强,用了unsigned long long,首先统计座位个数,奇数个是无法分配的直接回0;如果只有两个就回1;
对于大于两个且是偶数个的情况,能增加屏风的情况出现在两个座位 和 两个座位之间的植物之间,这一堆植物之间任意位置有且必须有一个地方要放屏风,如果这一堆有n个植物,则此处可诞生n+1种方案,各方案个数之间是相乘的关系(排列组合)。
所以需要统计每两个座位之间的植物个数cnt,用ans累乘(cnt+1),最后就是答案。
class Solution {
public:
int numberOfWays(string cor) {
unsigned long long ans = 1;
long long MOD = 1e9 + 7;
int Sn = count(cor.begin(), cor.end(), 'S');
if(Sn < 2 || Sn % 2) return 0;
if(Sn == 2) return 1;
int flag = 0;
int n = cor.size();
int cnt = 0;
for(int i = 0; i < n; i++) {
if(flag == 0 && cor[i] == 'S') flag = 1;
else if(flag == 1 && cor[i] == 'S') flag = 2;
else if(flag == 2 && cor[i] == 'S') {
ans = (ans % MOD) * ((cnt + 1) % MOD);
flag = 1;
cnt = 0;
}
else if(flag == 2 && cor[i] == 'P') cnt++;
}
ans %= MOD;
return (int)ans;
}
};