#335038 (C++14) No.334 門松ゲーム

• 問題: No.334 門松ゲーム
• ユーザー: matsukin1111
• 提出日時: 2019-04-03 13:19:35
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 5 ms / 2,000 ms
• コード長: 1,794 bytes
• コンパイル時間: 1,274 ms
• コンパイル使用メモリ: 98,592 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-05-10 03:24:22
• 合計ジャッジ時間: 2,043 ms
• ジャッジサーバーID （参考情報）: judge5 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	3 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	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	3 ms; 5,376 KB
testcase_11	AC	3 ms; 5,376 KB
testcase_12	AC	4 ms; 5,376 KB
testcase_13	AC	3 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	5 ms; 5,376 KB

ソースコード

#include<iostream>
#include<cstdio>
#include<cstring>
#include <cstdlib>  
#include <cmath>   
#include<cctype>
#include<string>
#include<set>
#include <map>
#include<algorithm>
#include <functional>
#include<vector>
#include<climits>
#include<stack>
#include<queue>
#include <deque>
#include <climits>
#include <typeinfo>
#include <utility> 
#define all(x) (x).begin(),(x).end()
#define rep(i,m,n) for(int i = m;i < n;++i)
#define pb push_back
#define fore(i,a) for(auto &i:a)
#define rrep(i,m,n) for(int i = m;i >= n;--i)
#define INF INT_MAX/2
using namespace std;
using ll = long long;
using R = double;
const ll MOD = 1e9 + 7;
const ll inf = 1LL << 50;
struct edge { ll from; ll to; ll cost; };

int n;
vector<int>k;
vector<int>ans;
int memo[1 << 12][2];

int check(int p,int q,int r) {
	vector<pair<int,int>>comp;
	comp.pb({k[p],p});
	comp.pb({k[q],q});
	comp.pb({k[r],r});
	sort(all(comp));

	if (comp[1].second == p || comp[1].second == r)return 1;
	else return 0;
}

int dfs(int used,int turn) {
	if (memo[used][turn] != -1)return memo[used][turn];

	int winner = 1 - turn;
	rep(p, 0, n)rep(q, p+1, n)rep(r, q+1, n) {
		if (!(used & (1 << p)) && !(used & (1 << q)) && !(used & (1 << r))) {
			if ((k[p] != k[q] && k[q] != k[r] && k[r] != k[p]) && check(p,q,r)) {
				int tmp = used;
				tmp += (1 << p) + (1 << q) + (1 << r);
				if (dfs(tmp, 1 - turn) == turn) {
					winner = turn;
					if (used == 0 && turn == 0) {
						ans.pb(p), ans.pb(q), ans.pb(r);
					}
					break;
				}
			}
		}
	}
	return memo[used][turn] = winner;
}


int main(){
	cin >> n;
	rep(i, 0, n) {
		int K;
		cin >> K;
		k.pb(K);
	}
	memset(memo, -1, sizeof(memo));

	if (dfs(0, 0) == 1)cout << -1 << endl;
	else {
		rep(i, 0, 3) {
			if (i)cout << " ";
			cout << ans[i];
		}
		cout << endl;
	}
	
	return 0;
}