ZOJ3818-Pretty Poem(暴力枚舉)

ZOJ3818-Pretty Poem(暴力枚舉)

題目鏈接

題意：求所給字符串是否符合ABABA或者ABABCAB的形式，如果可以的話輸出Yes，不可以的話為No。

思路：暴力枚舉A和B的長度，再用從長度減去3倍的AB長度，即為C的長度，看組合而成的字符串是否與給定的相等。在這裡string中的substr函數是個好東西。

代碼：

#include 
#include 
#include 
#include 
#include 

using namespace std;

char str[55];

int main() {
    int cas;
    scanf("%d", &cas);
    while (cas--) {
        scanf("%s", str); 
        string s;
        int len = strlen(str);
        for (int i = 0; i < len; i++)
            if (isupper(str[i]) || islower(str[i]))
                s += str[i];
        len = s.length();
        int flag = 0;
        for (int i = 1; i < len / 2 && !flag; i++) {
            for (int j = 1; j < len / 2 && !flag; j++) {
                string A = s.substr(0, i); 
                string B = s.substr(i, j); 
                if (A == B) 
                    continue;
                if (A + B + A + B + A == s) {
                    flag = 1;
                    break;
                }
                if (len - (i + j) * 3 > 0) {
                    string AB = A + B; 
                    string C = s.substr(2 * (i + j), len - (i + j) * 3); 
                    if (A == C || B == C)
                        continue;
                    if (AB + AB + C + AB == s) {
                        flag = 1; 
                        break; 
                    }
                } 
            } 
        } 
        if (flag)
            printf("Yes\n");
        else
            printf("No\n"); 
    }
    return 0;
}