Codeforces - 550A - Two Substrings

Question

题目大意

给你一个字符串 s ，要你判断是否有 AB 和 BA 同时出现在字符串中，但是不能连续，例如不能 ABA 这种。

解析

其实就是个简单的模拟题，但是还是错了好几次，还是菜。。。

给几个特殊样例:

ABABAB       YES
ABAXXXAB     YES
ABA          NO
ABAB         NO
ABABA        YES

处理就是当出现 ABA 或者 BAB 的时候，统计 AB 和 BA 都出现了，但是统计完不判断，而是跳过这三个字符。

import java.io.BufferedInputStream;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {

        Scanner cin = new Scanner(new BufferedInputStream(System.in));
        String str = cin.next();

        boolean isAB = false, isBA = false;
        for (int i = 0; i < str.length() - 1; i++) {
            if (str.charAt(i) == 'B' && str.charAt(i + 1) == 'A') {
                if (isAB) {
                    System.out.println("YES");
                    return;
                } else {
                    if(i < str.length()-2 && str.charAt(i+2) == 'B') { //BAB
                        isAB = true;
                        i += 2;// 跳过
                    }
                    isBA = true;
                }
            }
            if (str.charAt(i) == 'A' && str.charAt(i + 1) == 'B') {
                if (isBA ) {
                    System.out.println("YES");
                    return;
                } else {
                    if(i < str.length()-2 && str.charAt(i+2) == 'A') { // ABA
                        isBA = true;
                        i += 2; // 跳过
                    }
                    isAB = true;
                }
            }
        }
        System.out.println("NO");
    }
}