C. Prefixes and Suffixes

链接

[https://codeforces.com/contest/1092/problem/C]

题意

给你某个字符串的长度n,再给你2*n-2个前缀或者后缀

让你判断那些是前缀那些是后缀

关键是1到n-1的都有两个

分析

只需分析长度为n-1的那两个谁是前缀，谁是后缀

代码

#include<bits/stdc++.h>
using namespace std;
string s[210];
bool vis[110];
int main(){
	int n;
	//ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	//freopen("in.txt","r",stdin);
	while(~scanf("%d",&n)){
		bool flag=0;
		string s1="a",s2="b",pre;
		for(int i=0;i<2*n-2;i++)
		{
			cin>>s[i];
			if(s[i].size()==n-1&&!flag){
			s1=s[i]; flag=1;
			}
			else if(s[i].size()==n-1) s2=s[i];
		}
		int cnt=0;
		for(int i=0;i<2*n-2;i++)
			if(s1.substr(0,s[i].size())==s[i]&&s[i]!=s2) cnt++;
			//若前缀超过n-1,且s1的最长后缀等于s2的最长前缀，说明s1是前缀
		  if(cnt>=n-1&&s1.substr(1,s1.size()-1)==s2.substr(0,s2.size()-1))
            pre=s1;
        else pre=s2;//否则s2为前缀 

        memset(vis,0,sizeof(vis));//标记相同长度的是否已经是前缀
        for(int i=0;i<2*n-2;i++)
        if(s[i]==pre.substr(0,s[i].size())&&!vis[s[i].size()]) {
        	printf("P"); vis[s[i].size()]=1;
		}
        else printf("S");
        puts("");
	}
	return 0;
}