2022牛客寒假算法基础集训营4第一题 “R“

前言

限制条件下求字符串子串数问题。
博客链接http://www.retrogogogo.top/2022/02/08/algorithm/nowcoder/winter2022_4/

Problem A R

链接：https://ac.nowcoder.com/acm/contest/23479/A
来源：牛客网

题意

  小红拿到了一个长度为 $n$ 的字符串，该字符串仅由大写字母组成。 小红很喜欢红色（用’R’字母表示），但她非常讨厌紫色（用’P’字母表示）。她想取一个连续子串，该子串包含至少 $k$ 个’R’字符，且不能包含’P’字符。求方案数量。

方法解析

  双指针的思想，一个指针指向左端点，another指向右端点：枚举所有子串的左端点，那么对于一个左端点 $P_{left}$ ，所以满足条件的右端点就是 $P_{left}$ 后第一个’P’之前、第 $k$ 个’ R’之后的所有点。我们定义两个数组： $nextP[200010]$ 和 $nextkR[200010]$，其中 $nextP[i]$ 表示 $i$ 之后第一个’P’的位置， $ nextkR[i]$ 表示 $i$ 之后第 $k$ 个’R’的位置.（当然是找得到的情况下）。那么右端点一定是单调递增的，每一次 $P_{left}$ 的遍历都能得到一个 $nextP[i]-nextkR[i]$ 的答案累加。

  注：为了方便编程，我做了s="P"+s; 操作，代码不怎么精炼但是逻辑理顺后写起来还是很清晰的。（答案范围ll切记）

参考代码

typedef long long ll;
int nextP[200010];
int nextkR[200010];
void solveA()
{
    int n,k;
    cin>>n>>k;
    string s;
    cin>>s;
    s="P"+s;
    ll ans = 0;
    for(int i=1;i<=n;i++){
        if(s[i-1]=='P'){
            int cntR = 0;
            for(int j=i;j<=n;j++){
                if(s[j]=='R'){
                    cntR++;
                    if(cntR==k){
                        nextkR[i] = j;
                    }
                }
                else if(s[j]=='P'){
                    nextP[i] = j;
                    break;
                }
                
            }
            if(nextP[i]==0){
                nextP[i] = n+1;
            }
            if(nextkR[i]==0){
                i=nextP[i];
            }
            else{
                ans+=nextP[i]-nextkR[i];
            }
        }
        else if(s[i-1]=='R'){
            nextP[i] = nextP[i-1];
            for(int j=nextkR[i-1]+1;j<nextP[i-1];j++){
                if(s[j]=='R'){
                    nextkR[i] = j;
                    break;
                }
            }
            if(nextkR[i]==0){
                i=nextP[i];
            }
            else{
                ans+=nextP[i]-nextkR[i];
            }
        }
        else{
            nextP[i] = nextP[i-1];
            nextkR[i] = nextkR[i-1];
            if(nextkR[i]!=0){
                ans+=nextP[i]-nextkR[i];
            }
        }
    }
    cout<<ans<<endl;
    

}