D. Journey（1月29日晚div2 D题）

题目链接

D. Journey

有n+1个城市。n条有向边，连接 i -> i+1。然后现在商人以一个点为起点开始遍历。每移动一次，所有的边就会调转方向。问以所有i为起点。能访问到的城市的数量。

分析：

从一个点往左走的最远距离，是LR字母交替的数量，同理向右走的最远距离一样。

我们可以开两个数组sum1,sum2，分别记录从当前点向左，向右最远能走多少，res数组记录每个点的最终答案

用递推的方式可以求出sum1，sum2数组，再根据这两个数组更新res数组

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+5;
int sum1[N],sum2[N],res[N];
//sum1¼Ç¼ÿ¸öµãÏò×ó×ß×î¶àÄÜ×߶àÔ¶£¬°üÀ¨×Ô¼º 
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		int n;
		string s;
		cin>>n>>s;
		sum1[0]=1;
		for(int i=1;i<n;i++){
			if(s[i-1]==s[i]) sum1[i]=1;
			else sum1[i]=sum1[i-1]+1;
		}
		sum2[n-1]=1;
		for(int i=n-2;i>=0;i--){
			if(s[i+1]==s[i]) sum2[i]=1;
			else sum2[i]=sum2[i+1]+1;
		}
		for(int i=1;i<=n-1;i++){
			res[i]=1;
			if(s[i-1]=='L') res[i]+=sum1[i-1];
			if(s[i]=='R') res[i]+=sum2[i];
		}
		if(s[0]=='R') res[0]=sum2[0]+1;
		else res[0]=1;
		if(s[n-1]=='L') res[n]=sum1[n-1]+1;
		else res[n]=1;
		
		for(int i=0;i<=n;i++){
			cout<<res[i]<<" ";
		}
		cout<<endl;
	}
}