結果

問題 No.2240 WAC
ユーザー shobonvip
提出日時 2023-02-09 03:50:42
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,602 bytes
コンパイル時間 2,318 ms
コンパイル使用メモリ 207,020 KB
最終ジャッジ日時 2025-02-10 11:44:54
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 13 TLE * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include<atcoder/maxflow>
using namespace std;
using namespace atcoder;

// importbisect
template <typename T>
int bisect_left(vector<T> &X, T v){
	return lower_bound(X.begin(), X.end(), v) - X.begin();
}

template <typename T>
int bisect_right(vector<T> &X, T v){
	return upper_bound(X.begin(), X.end(), v) - X.begin();
}
// -----

int main(){
	int n, m; cin >> n >> m;
	string s; cin >> s;
	int k = n + m;
	mf_graph<int> mf(4*k+2);

	vector<int> w(0);
	vector<int> a(0);
	vector<int> c(0);
	for (int i=0; i<2*k; i++){
		if (s[i] == 'W') w.push_back(i);
		if (s[i] == 'A') a.push_back(i);
		if (s[i] == 'C') c.push_back(i);
	}
	
	// スタートから各 W, C へ
	for (int i=0; i<k; i++){
		mf.add_edge(4*k, i, 1);
	}

	// 各 A からゴールへ
	for (int i=0; i<k; i++){
		mf.add_edge(3*k+i, 4*k+1, 1);
	}

	// ベルトコンベアー (前へ)
	for (int i=0; i<k-1; i++){
		mf.add_edge(k+i, k+i+1, 300000);
	}

	// ベルトコンベアー (うしろへ)
	for (int i=0; i<k-1; i++){
		mf.add_edge(2*k+i+1, 2*k+i, 300000);
	}

	// ベルトコンベアーから各 A へ
	for (int i=0; i<k; i++){
		mf.add_edge(k+i, 3*k+i, 1);
		mf.add_edge(2*k+i, 3*k+i, 1);
	}

	// 各 W からベルトコンベアへ
	for (int i=0; i<n; i++){
		int targ = bisect_right(a, w[i]);
		if (targ < k) mf.add_edge(i, k+targ, 1);
	}

	// 各 C からベルトコンベアへ
	for (int i=0; i<m; i++){
		int targ = bisect_left(a, c[i]);
		targ--;
		if (targ >= 0) mf.add_edge(i+n, 2*k+targ, 1);
	}

	if (mf.flow(4*k, 4*k+1) == k){
		cout << "Yes" << endl;
	}else{
		cout << "No" << endl;
	}
	
}
0