Codeforces1092C STL

https://codeforces.com/contest/1092/problem/C
集合count()函数
字符串empty() clear()函数
注意check不一定第一次就成功 每次都要置空ans="";

题意:
对长度为n的目标串 给2*n-2个它的子串
长度为1-n-1的前缀各一个
长度为1-n-1的后缀各一个

#include <bits/stdc++.h>
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define pb push_back
#define mp make_pair
#define IO ios::sync_with_stdio(false)
#define fi first
#define se second
using namespace std;
const int maxn=1005;
const int INF=0x3f3f3f3f;
typedef long long ll;
typedef pair<int,int> pii;


int n,m,q,cnt,f,cha,res,cnt1;

string s[305],a,b,ans;	//长度为1-n-1的前缀 长度为1-n-1的后缀 99个+99个 
bool check(string t){	//构造长度为n的字符串 4种情况 
	set<string> pre,suf;
	string x="";
	ans="";
	for(int i=0;i<t.size()-1;i++){
		x+=t[i];
		pre.insert(x);
	}
	x.clear();
	for(int i=t.size()-1;i>=1;i--){
		x=t[i]+x;
		suf.insert(x);
	}
	for(int i=0;i<2*n-2;i++){
		if(pre.count(s[i])){	//计数 
			ans+='P';
			pre.erase(s[i]);
		}
		else if(suf.count(s[i])){
			ans+='S';
			suf.erase(s[i]);
		}
		else
			return 0;	//pre suf中不存在这样的前缀后缀 构造的串有问题 
	}
	return 1;
}
int main(){
	cin>>n;
	for(int i=0;i<2*n-2;i++){
		cin>>s[i];
	}
	for(int i=0;i<2*n-2;i++){
		if(s[i].size()==n-1){
			if(a.empty())
				a+=s[i];	//先压入的长度为n-1的串 可能是前缀 、后缀 
			else
				b+=s[i];
		}
	}
	if(check(a.front()+b))
		cout<<ans<<endl;
	else if(check(a+b.back()))
		cout<<ans<<endl;
	else if(check(b.front()+a))
		cout<<ans<<endl;
	else if(check(b+a.back()))
		cout<<ans<<endl;
		
	return 0;
}

将长度为n-1的2个串拿出来 1个前缀 1个后缀
构造有4种情况
a+b a[0]+b a+b[n-2]
b+a b[0]+a b+a[n-2]
正向扫长度1-n-1的前缀 压入
反向扫长度1-n-1的后缀 压入
验证输入的2*n-2个字符串是否在前缀/后缀集合中 否则构造错误