結果

提出番号 1375
提出者 E869120
言語 C++
提出日時 2018-07-31 11:42:10
問題名 (64)Or Plus Max 2
結果 AC
点数 100%

テストケース

テストケース 結果 得点 実行時間 メモリ使用量
1 AC 100% 2ms 8400KB
2 AC 100% 2ms 8416KB
3 AC 100% 1ms 7792KB
4 AC 100% 2ms 8352KB
5 AC 100% 2ms 8064KB
6 AC 100% 2ms 8368KB
7 AC 100% 2ms 7968KB
8 AC 100% 2ms 8688KB
9 AC 100% 2ms 8688KB
10 AC 100% 2ms 8288KB
11 AC 100% 2ms 7216KB
12 AC 100% 2ms 7600KB
テストケース 結果 得点 実行時間 メモリ使用量
13 AC 100% 4ms 10656KB
14 AC 100% 4ms 10288KB
15 AC 100% 4ms 9280KB
16 AC 100% 3ms 8864KB
17 AC 100% 3ms 11008KB
18 AC 100% 3ms 9072KB
19 AC 100% 3ms 8688KB
20 AC 100% 3ms 7520KB
テストケース 結果 得点 実行時間 メモリ使用量
21 AC 100% 6ms 12816KB
22 AC 100% 21ms 14560KB
23 AC 100% 53ms 16112KB
24 AC 100% 33ms 13792KB
25 AC 100% 58ms 18144KB
26 AC 100% 8ms 9344KB
27 AC 100% 78ms 19968KB
28 AC 100% 4ms 8528KB
29 AC 100% 10ms 8720KB
30 AC 100% 14ms 9472KB
テストケース 結果 得点 実行時間 メモリ使用量
31 AC 100% 1446ms 53264KB
32 AC 100% 728ms 34768KB
33 AC 100% 1648ms 57424KB
34 AC 100% 1360ms 52112KB
35 AC 100% 487ms 44512KB
36 AC 100% 233ms 27280KB
37 AC 100% 4ms 8400KB
38 AC 100% 121ms 17424KB
39 AC 100% 103ms 8160KB
40 AC 100% 1038ms 43936KB
41 AC 100% 1328ms 54896KB

ソースコード

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

vector<pair<int, long long>>x[1024];
int 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++) {
		int SS = 0;
		for (int j = i; j < N; j++) SS += min(j + 1, S);
		for (int j = max(0, K - SS); j <= K; j++) {
			for (int k = 0; k < 1024; k++) {
				if (dp[i][j][k] == 0) continue;
				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] += (1LL * dp[i][j][k] * cost) % mod;
					if (dp[i + 1][j + F[to]][to] >= mod) 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;
}