首页 > 代码库 > Codeforces Round #306 (Div. 2) A

Codeforces Round #306 (Div. 2) A

题意

给一个字符串(长度<=10^5)。问当中有没有一个”BA”和一个”AB”呢?假设都有而且它们不反复(即ABA不算),输出YES。否则输出NO。

思路

一開始想简单了…..
我们扫一遍,把全部”AB”字符串中A的索引放入一个vector a,把全部”BA”字符串中B的索引放入还有一个vector b。最后扫一遍两个vector。假设发现一个b的值既不是一个a的值+1,也不是那个a的值-1,那么肯定就存在不反复的”BA”和”AB”了。

代码

#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int maxn = 100010;
char s[maxn];
vector<int> a;
vector<int> b;
int main()
{
    scanf("%s",s);
    int len = strlen(s);
    bool flag1 = false;
    bool flag2 = false;
    for(int i = 0 ; i < len ; i ++) {
        if(s[i] == ‘A‘ && s[i+1] == ‘B‘) {
            a.push_back(i);
        }
        if(s[i] == ‘B‘ && s[i+1] == ‘A‘) {
            b.push_back(i);
        }
    }
    for(int i = 0 ; i < a.size() ; i ++) {
        for(int j = 0 ; j < b.size() ; j ++) {
            if(a[i]!=b[j]+1 && a[i]!=b[j]-1) {
                printf("YES\n");
                return 0;
            }
        }
    }
    printf("NO\n");
    return 0;
}
<script type="text/javascript"> $(function () { $(‘pre.prettyprint code‘).each(function () { var lines = $(this).text().split(‘\n‘).length; var $numbering = $(‘
    ‘).addClass(‘pre-numbering‘).hide(); $(this).addClass(‘has-numbering‘).parent().append($numbering); for (i = 1; i <= lines; i++) { $numbering.append($(‘
  • ‘).text(i)); }; $numbering.fadeIn(1700); }); }); </script>

Codeforces Round #306 (Div. 2) A