#305825 (C++14) No.775 tatyamと素数大富豪(hard)

• 問題: No.775 tatyamと素数大富豪(hard)
• ユーザー: kriii
• 提出日時: 2018-12-22 01:50:29
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 44 ms / 2,000 ms
• コード長: 1,067 bytes
• コンパイル時間: 476 ms
• コンパイル使用メモリ: 47,640 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-22 02:49:41
• 合計ジャッジ時間: 1,304 ms
• ジャッジサーバーID （参考情報）: judge2 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	3 ms; 5,376 KB
testcase_07	AC	5 ms; 5,376 KB
testcase_08	AC	6 ms; 5,376 KB
testcase_09	AC	23 ms; 5,376 KB
testcase_10	AC	22 ms; 5,376 KB
testcase_11	AC	1 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	3 ms; 5,376 KB
testcase_14	AC	44 ms; 5,376 KB

ソースコード

#include <stdio.h>
#include <algorithm>
using namespace std;

int N,K,L,A[202],C[100],U[100],X;

bool good(int a, int r, int l = 0)
{
	if (X < a) X = a;
	if ((L - a) / 2 < r) return false;
	if (a == L) return true;
	if (a + 2 <= L && A[a]){
		int u = A[a] * 10 + A[a+1];
		if (u != l * 11){
			int &v = U[u];
			if (v < C[u]){
				v++;
				if (good(a+2, r-1, u)) return true;
				v--;
			}
		}
	}
	int &v = U[A[a]];
	if (v < C[A[a]]){
		v++;
		if (good(a+1, r, A[a])) return true;
		v--;
	}
	return false;
}

int main()
{
#ifdef __LOCAL
	freopen ("input.txt","r",stdin);
#endif

	scanf ("%d %d",&N,&K);
	for (int i=0;i<N;i++){
		int x; scanf ("%d",&x); C[x]++;
		if (x < 10) A[L++] = x;
		else A[L++] = x / 10, A[L++] = x % 10;
	}
	sort(A,A+L);
	reverse(A,A+L);

	int rem = 0;
	for (int i=10;i<100;i++) rem += C[i];

	do{
		for (int i=0;i<100;i++) U[i] = 0;
		X = 0;
		if (good(0,rem)){
			for (int i=0;i<L;i++) printf ("%d",A[i]);
			puts("");
			if (--K == 0) break;
		}
		else{
			if (X + 2 < L) sort(A+X+2,A+L);
		}
	}while(prev_permutation(A,A+L));

	return 0;
}