結果

提出番号 1352
提出者 E869120
言語 C++
提出日時 2018-07-20 15:41:26
問題名 (64)Or Plus Max 2
結果 TLE
点数 78%

テストケース

テストケース 結果 得点 実行時間 メモリ使用量
1 AC 100% 2ms 8400KB
2 AC 100% 2ms 8048KB
3 AC 100% 2ms 7472KB
4 AC 100% 2ms 7968KB
5 AC 100% 2ms 8400KB
6 AC 100% 2ms 8400KB
7 AC 100% 2ms 8672KB
8 AC 100% 2ms 7856KB
9 AC 100% 2ms 8416KB
10 AC 100% 1ms 8288KB
11 AC 100% 2ms 8048KB
12 AC 100% 2ms 7968KB
テストケース 結果 得点 実行時間 メモリ使用量
13 AC 100% 6ms 19648KB
14 AC 100% 7ms 23104KB
15 AC 100% 8ms 28480KB
16 AC 100% 5ms 14912KB
17 AC 100% 6ms 22336KB
18 AC 100% 4ms 15616KB
19 AC 100% 26ms 83008KB
20 AC 100% 24ms 77648KB
テストケース 結果 得点 実行時間 メモリ使用量
21 AC 100% 12ms 29632KB
22 AC 100% 68ms 42928KB
23 AC 100% 157ms 39152KB
24 AC 100% 75ms 22064KB
25 AC 100% 195ms 47136KB
26 AC 100% 19ms 14384KB
27 AC 100% 151ms 38192KB
28 AC 100% 8ms 12320KB
29 AC 100% 475ms 106528KB
30 AC 100% 269ms 70064KB
テストケース 結果 得点 実行時間 メモリ使用量
31 AC 100% 2019ms 174624KB
32 AC 100% 1291ms 102864KB
33 AC 100% 3358ms 282752KB
34 AC 100% 3027ms 255664KB
35 AC 100% 929ms 168784KB
36 AC 100% 373ms 81232KB
37 AC 100% 11ms 16912KB
38 TLE 0% 6260ms 469008KB
39 AC 100% 121ms 8416KB
40 AC 100% 4934ms 366384KB
41 AC 100% 4551ms 342288KB

ソースコード

#include <iostream>
#include <vector>
#include <string>
using namespace std;
#pragma warning (disable: 4996)

vector<pair<int, long long>>x[1024];
long long N, K, P, S, F[1024], dp[44][444][1024], dp2[44][44], mod = 1000000007;

vector<int>calc(int pos) {
	if (pos == 0) { return vector<int>{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}; }
	int len = 0, sz = 1;
	while (sz * 2 <= pos) { sz *= 2; len++; }

	int t = pos - sz, cx = 0; vector<int>A(11, 0); A[0]++;
	for (int i = 0; i < len; i++) {
		if ((t / (1 << i)) % 2 == 1) cx++;
		A[cx]++;
	}
	return A;
}

long long ncr(int n, int r) {
	if (n < 0 || n < r) return 0;
	return dp2[n - r][r];
}

void init() {
	for (int i = 0; i < 44; i++) {
		for (int j = 0; j < 44; j++) {
			if (i == 0 || j == 0) dp2[i][j] = 1;
			else dp2[i][j] = dp2[i - 1][j] + dp2[i][j - 1];
		}
	}
	for (int i = 0; i < P; i++) {
		vector<int>L1 = calc(i);
		for (int j = 0; j < L1.size(); j++) { if (L1[j] == 0) { F[i] = j; break; } }
		for (int j = 1; j < P; j++) {
			vector<int>L2 = calc(j); bool OK = false;
			for (int k = 0; k < 10; k++) {
				if (L1[k] != L2[k + 1] && L2[k + 1] != 0) OK = true;
			}

			if (OK == true) continue;

			long long res = 0, LA = 0, LB = 0, s1 = S;
			for (int k = 0; k < 10; k++) { if (L2[k + 1] == 0) { if (LA == 0) LA = L1[k]; else LB += L1[k]; } }
			for (int k = 0; k <= 10; k++) s1 -= L1[k];

			for (int k = 0; k <= LA; k++) {
				// k : LA の値
				if (LA >= 1 && k == 0) continue;
				for (int l = 0; l <= LB; l++) {
					// l : LB の値
					if (k + l > L2[0]) break;
					res += ncr(LA, k) * ncr(LB, l) * ncr(s1, L2[0] - k - l);
				}
			}
			x[i].push_back(make_pair(j, res));
		}
	}
}

int main() {
	cin >> N >> K >> P; for (int i = 0; i < 11; i++) { if ((1 << i) == P) S = i; }
	init();
	if (K > 360) { cout << "0" << endl; return 0; }

	dp[0][0][0] = 1;
	for (int i = 0; i < N; i++) {
		for (int j = 0; j <= K; j++) {
			for (int k = 0; k < 1024; k++) {
				for (int l = 0; l < x[k].size(); l++) {
					long long to = x[k][l].first, cost = x[k][l].second;
					dp[i + 1][j + F[to]][to] += dp[i][j][k] * cost;
					dp[i + 1][j + F[to]][to] %= mod;
				}
			}
		}
	}
	long long res = 0;
	for (int i = 0; i < 1024; i++) res += dp[N][K][i];
	cout << res % mod << endl;
	return 0;
}