CF 1476 D. Journey(线性dp)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int dp[300050][2][2];
string cc;
void solve(){
    
    
    int n;
    cin>>n;
    for(int i=0;i<=n;i++)dp[i][0][0]=dp[i][0][1]=dp[i][1][0]=dp[i][1][1]=0;
    cin>>(cc);
    dp[n][1][0]=dp[n][1][1]=0;
    dp[0][0][0]=dp[0][0][1]=0;
    for(int i=n-1;i>=0;i--)
    {
    
    
        if(cc[i]=='R')dp[i][1][0]=dp[i+1][1][1]+1;
        if(cc[i]=='L')dp[i][1][1]=dp[i+1][1][0]+1;
    }
    for(int i=1;i<=n;i++)
    {
    
    
        if(cc[i-1]=='L')dp[i][0][0]=dp[i-1][0][1]+1;
        if(cc[i-1]=='R')dp[i][0][1]=dp[i-1][0][0]+1;
    }
    for(int i=0;i<=n;i++)cout<<dp[i][1][0]+dp[i][0][0]+1<<' ';
    cout<<endl;
}
int main()
{
    
    
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int T;
    cin>>T;
    while(T--)solve();
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Ambrumf/article/details/121129272