【leectode 2022.2.8】分隔长廊的方案数

在一个图书馆的长廊里，有一些座位和装饰植物排成一列。给你一个下标从 0 开始，长度为 n 的字符串 corridor ，它包含字母 ‘S’ 和 ‘P’ ，其中每个 ‘S’ 表示一个座位，每个 ‘P’ 表示一株植物。

在下标 0 的左边和下标 n - 1 的右边 已经 分别各放了一个屏风。你还需要额外放置一些屏风。每一个位置 i - 1 和 i 之间（1 <= i <= n - 1），至多能放一个屏风。

请你将走廊用屏风划分为若干段，且每一段内都 恰好有两个座位 ，而每一段内植物的数目没有要求。可能有多种划分方案，如果两个方案中有任何一个屏风的位置不同，那么它们被视为 不同 方案。

请你返回划分走廊的方案数。由于答案可能很大，请你返回它对 109 + 7 取余 的结果。如果没有任何方案，请返回 0 。

示例 1：

输入：corridor = “SSPPSPS”
输出：3
解释：总共有 3 种不同分隔走廊的方案。
上图中黑色的竖线表示已经放置好的屏风。
上图每种方案中，每一段都恰好有 两个 座位。
示例 2：

输入：corridor = “PPSPSP”
输出：1
解释：只有 1 种分隔走廊的方案，就是不放置任何屏风。
放置任何的屏风都会导致有一段无法恰好有 2 个座位。
示例 3：

输入：corridor = “S”
输出：0
解释：没有任何方案，因为总是有一段无法恰好有 2 个座位。

提示：

n == corridor.length
1 <= n <= 105
corridor[i] 要么是 ‘S’ ，要么是 ‘P’ 。

java代码：

class Solution {
    int mod=(int)1e9+7;
    public int numberOfWays(String corridor) {
        int count=0;
        for(int i=0;i<corridor.length();i++){if(corridor.charAt(i)=='S'){count++;}}
        if(count==0||count%2==1){return 0;}
        long ans=1;
        int i=0;
        while(corridor.charAt(i)=='P'){i++;}//i定位于第一个S
        while(count>2){
            int j=i+1,p=0;//p记录S之间的P
            while(corridor.charAt(j)=='P'){j++;}//j定位于后一个S
            j++;//定位于后一个S的后边
            while(corridor.charAt(j)=='P'){
                j++;
                p++;
            }
            ans=(ans*(p+1))%mod;
            count-=2;
            i=j;
        }
        return (int)ans;
    }
}