#110515 (C++11) No.412 花火大会

• 問題: No.412 花火大会
• ユーザー: startcpp
• 提出日時: 2016-08-12 23:49:36
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 2 ms / 2,000 ms
• コード長: 1,060 bytes
• コンパイル時間: 441 ms
• コンパイル使用メモリ: 58,748 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-25 06:27:33
• 合計ジャッジ時間: 1,127 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

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

ソースコード

#include <iostream>
#include <algorithm>
using namespace std;

int w[3];
int n;
int e[30];
int dp[31][3];	//dp[i][j] … i枚のレジャーシートについて「使う, 使わない」を決めていて、j家族にシートを渡せる状態. そこからの選び方.

int power(int a, int n) {
	int ret = 1;
	for (int i = 0; i < n; i++) ret *= a;
	return ret;
}

int dfs(int i, int j) {
	if (j == 3) return power(2, n - i);
	if (i == n) return 0;
	if (dp[i][j] != -1) return dp[i][j];
	
	int ret = 0;
	//レジャーシートiを使わない
	ret += dfs(i + 1, j);
	//使う
	ret += dfs(i + 1, j + (e[i] >= w[j]));
	return (dp[i][j] = ret);
}

int main() {
	int i, j;
	
	cin >> w[0] >> w[1] >> w[2];
	cin >> n;
	for (i = 0; i < n; i++) cin >> e[i];
	
	//ソートしても答えは変わらないので、ソートする。→家族0,1,2の順でレジャーシートを割り当てればよくなる。
	sort(w, w + 3);
	sort(e, e + n);
	
	for (int i = 0; i <= n; i++) for (int j = 0; j < 3; j++) dp[i][j] = -1;
	cout << dfs(0, 0) << endl;
	return 0;
}