#253086 (C++11) No.672 最長AB列

• 問題: No.672 最長AB列
• ユーザー: ohreitetsu
• 提出日時: 2018-04-22 17:49:08
• 言語: C++11 (gcc 11.4.0)
• 結果: WA
• 実行時間: -
• コード長: 1,308 bytes
• コンパイル時間: 607 ms
• コンパイル使用メモリ: 63,480 KB
• 実行使用メモリ: 12,092 KB
• 最終ジャッジ日時: 2024-06-27 20:29:06
• 合計ジャッジ時間: 4,271 ms
• ジャッジサーバーID （参考情報）: judge2 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 10,624 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	1 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	WA	-
testcase_09	AC	6 ms; 5,376 KB
testcase_10	AC	6 ms; 5,376 KB
testcase_11	AC	5 ms; 5,376 KB
testcase_12	TLE	-
testcase_13	--	-
testcase_14	--	-
testcase_15	--	-
testcase_16	--	-
testcase_17	--	-
testcase_18	--	-

ソースコード

#include <iostream>
#include <vector>
#include <string> 
using namespace std;
void setLens(string S,vector<pair<int,int>> &lenVector)
{
	int i;
	int maxLen=0;
	string maxString;
	int aNum=0,bNum=0;
	int len=1;
	for (i=0;i<S.size();i++){
		if (S[i]=='A'){
			aNum++;
		}else{
			bNum++;
		}
		pair<int,int> num;
		num.first=aNum;
		num.second=bNum;
		lenVector.push_back(num);
	}
}
int GetMaxLen(vector<pair<int,int>> lenVector,char ch){
	int i;
	int MaxLen=0;
	for (i=0;i<lenVector.size();i++){
		if (ch=='A'){
			lenVector[i].first--;
		}else if (ch=='B'){
			lenVector[i].second--;
		}
		if (lenVector[i].first==lenVector[i].second){
			if (MaxLen<2*lenVector[i].first){
				MaxLen=2*lenVector[i].first;
			}
		}
		lenVector.erase(lenVector.begin());
	}
	return MaxLen;
}
int main(int argc, char* argv[])
{
	string S;
	cin>>S;
	int len=S.size();
	int aNum=0,bNum=0;
	int i=0;
	for (i=0;i<len;i++){
		if (S[i]=='A'){
			aNum++;
		}else{
			bNum++;
		}
	}
	if (aNum==0 || bNum==0){
		cout<<0<<endl;
		return 0;
	}else if (aNum==bNum){
		cout<<len<<endl;
		return 0;
	}
	vector<pair<int,int>> lenVector;
	setLens(S,lenVector);
	int MaxLen=GetMaxLen(lenVector,' ');		
	for (i=0;i<len;i++){
		int maxLen=GetMaxLen(lenVector,S[i]);
		if (MaxLen<maxLen){
			MaxLen=maxLen;
		}
	}
	cout<<MaxLen<<endl;
	return 0;
}