Codeforces Round #306 (Div. 2) A
阿新 • • 發佈:2017-07-22
return code add stl ext scan java pac 一個
題意
給一個字符串(長度<=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;
}
Codeforces Round #306 (Div. 2) A