Codeforces Round #527 (Div. 3) C. Prefixes and Suffixes

Codeforces Round #527 (Div. 3) C. Prefixes and Suffixes

题目

       题目很简单。有一个长为n的字符串，现给出2n-2个前缀和 和 后缀和（长度分别从1到n-1），判断给出的每个字符串是前缀和还是后缀和。要求前缀和和后缀和的个数一样。
       保证从1到n个−1的每个长度正好有2个字符串。它还保证了这些字符串是一些现有长度为n的字符串的前缀和后缀。

       思路很简单。找到长度为n-1的两个子串，能构造出两个符合的原串：子串1的最后一个字母+子串2、子串1+子串2的最后一个字母。对每一个给出的子串，判断它是否前缀和和后缀和，将1表示子串是前缀or后缀和，0表示不是，那么有4种情况：
       1、11：
       2、10：
       3、01：
       4、00：
       考码力的一道题。

我的代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 205;
int a[maxn];
int n, t, k;
string str[maxn];
string ss[3];
string ans;
int flag[maxn];
int check(int k){
    for(int i = 1; i <= 2*n-2; i++){
        if(ss[k].find(str[i]) == 0){
            reverse(ss[k].begin(), ss[k].end());
            reverse(str[i].begin(), str[i].end());
            if(ss[k].find(str[i]) == 0){ //11
                if(flag[str[i].size()])   { ans += "S"; }
                else{
                    flag[str[i].size()] = 1;
                    ans += "P";
                }
            }
            else{
                ans += "P"; //10
            }
            reverse(ss[k].begin(), ss[k].end());
            reverse(str[i].begin(), str[i].end());
        }
        else{
            reverse(ss[k].begin(), ss[k].end());
            reverse(str[i].begin(), str[i].end());
            if(ss[k].find(str[i]) == 0){
                ans += "S"; //01
            }
            else{//00
                reverse(ss[k].begin(), ss[k].end());
                reverse(str[i].begin(), str[i].end());
                return 1;
            }
            reverse(ss[k].begin(), ss[k].end());
            reverse(str[i].begin(), str[i].end());
        }
    }
    return 0;
}

int main(){
    cin >> n;
    string s1[3], s2[3];
    int num1, num2;
    num1 = num2 = 1;
    for(int i = 1; i <= 2*n-2; i++){
        cin >> str[i];
        flag[i] = 0;
        if(str[i].size() == n-1){s2[num1++] = str[i]; }
    }

    ss[1] = s2[1] + s2[2][n-2];
    ss[2] = s2[2] + s2[1][n-2];
    
    if(check(1)) {
        ans.clear();
        memset(flag, 0, sizeof(flag));
        check(2);
        cout<<ans<<endl;
     }
    else{ cout<<ans<<endl; }
    return 0;
}

wa点

1.判断是否后缀时，除了要把str翻转，还要把substr翻转；
2.每次reverse完了之后还要再reverse回来；
3.若不符合要求，ans要清空，flag要重新初始化；
4.最后！！！！！wa点！！！！！对于string来说，s = s1 + s2 和 s = s2 + s1 是不同的！！！！！

参考代码

#include <algorithm>
#include <iostream>
#include <map>
#include <set>
#include <queue>
#include <cmath>
#include <stack>
#include <cstring>
#include <queue>
using namespace std;
#define ll long long
#define ull unsigned long long
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define IO ios::sync_with_stdio(false);cin.tie(0)
const int maxn = 1005;
const int mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;

inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
/*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*/

int n;
string str[205];
char ans[205];
bool check1(int k,string now){
    for(int i = 0;i < str[k].size();i ++){
        if(str[k][i] != now[i]) return false;
    }
    return true;
}
bool check2(int k,string now){
    for(int i = now.size() - 1,j = str[k].size() - 1;j >= 0;i --,j --){
        if(str[k][j] != now[i]) return false;
    }
    return true;
}

bool x(string now1){
    set<string>s;
    for(int i = 1;i <= 2*n - 2;i ++){
        int flag = 0,p1 = 0,p2 = 0;
        if(check1(i,now1)) p1 = 1;
        if(check2(i,now1)) p2 = 1;
        if(p1 && p2){ //11
            if(s.find(str[i]) != s.end()) ans[i] = 'S';
            else ans[i] = 'P';
            s.insert(str[i]);
        }
        else if(p1 && !p2){ // 10
            ans[i] = 'P';
            s.insert(str[i]);
        }
        else if(p2 && !p1){ // 01
            ans[i] = 'S';
            s.insert(str[i]);
        }
        else return false;//00
    }
    return true;
}
string str1,str2;
int main() {

    cin >> n;
    int flag1 = 0;
    for(int i = 1;i <= 2*n - 2;i ++){
        cin >> str[i];
        if(str[i].size() == n - 1 ) {
            if(!flag1){
                 str1 = str[i];  flag1 = 1;
            }
            else{
                str2 = str[i];
            }

        }
    }
    string now1 = str1 + str2[n - 2];
    string now2 = str2 + str1[n - 2];

    if(x(now1)){
        for(int i = 1;i <= 2*n - 2;i ++) cout << ans[i];
    }
    else {
        int b = x(now2);
        for(int i = 1;i <= 2*n - 2;i ++) cout << ans[i];
    }
    return 0;
}