結果

問題 No.941 商とあまり
ユーザー QCFium
提出日時 2019-10-31 14:17:40
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,959 bytes
コンパイル時間 2,746 ms
コンパイル使用メモリ 200,788 KB
実行使用メモリ 17,464 KB
最終ジャッジ日時 2024-09-14 22:04:02
合計ジャッジ時間 10,720 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 6
other TLE * 1 -- * 103
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

int ri() {
	int n;
	scanf("%d", &n);
	return n;
}

std::string solve(int n, int x, const std::vector<int> &a_) {
	auto a = a_;
	if (n == 1) {
		std::string s(x, '0');
		if (a[0] <= x) s[a[0] - 1] = '1';
		return s;
	}
	std::sort(a.begin(), a.end());
	int64_t prod = 1;
	for (int i = 0; i < n; i++) {
		prod *= a[i] + 1;
		if (prod > x + 1) return std::string(x, '0');
	}
	if (std::count(a.begin(), a.end(), 1)) {
		std::string res(x, '0');
		for (int i = prod - 1; i <= x; i++) res[i - 1] = '1';
		return res;
	}
	
	std::map<int, int> nums;
	for (auto i : a) nums[i]++;
	std::vector<std::pair<int, int> > all;
	for (auto i : nums) all.push_back(i);
	std::map<std::vector<int>, std::vector<bool> > dp;
	int m = nums.size();
	for (int i = 0; i < m; i++) {
		std::vector<int> num(all.size());
		num[i] = 1;
		int size = x + 1;
		for (int j = 0; j < m; j++) for (int k = 0; k < all[j].second - num[j]; k++) size /= all[j].first;
		std::vector<bool> val(size);
		assert(all[i].first < size);
		val[all[i].first] = true;
		dp[num] = val;
	}
	for (auto &i : dp) {
		/*
		std::cerr << "key:";
		for (auto j : i.first) std::cerr << j << " ";
		std::cerr << std::endl << "val:";
		for (auto j : i.second) std::cerr << j << " ";
		std::cerr << std::endl;*/
		for(auto &j : dp) {
			bool cont = true;
			std::vector<int> new_num(m);
			for (int k = 0; k < m; k++) if ((new_num[k] = i.first[k] + j.first[k]) > all[k].second) {
				cont = false;
				break;
			}
			if (!cont) continue;
			
			int new_size = x + 1;
			for (int k = 0; k < m; k++) for (int l = 0; l < all[k].second - new_num[k]; l++) new_size /= all[k].first;
			std::vector<bool> new_val(new_size);
			for (int k = 1; k < (int) i.second.size(); k++) {
				if (!i.second[k]) continue;
				std::vector<bool> tmp(new_size);
				for (int l = 1; l < (int) j.second.size(); l++) {
					if (!j.second[l]) continue;
					int64_t base = (int64_t) (k + 1) * (l + 1) - 1;
					if (base < new_size) tmp[base] = true;
				}
				for (int l = 0; l < new_size - k; l++) if (tmp[l]) tmp[l + k] = true;
				for (int l = 0; l < new_size; l++) if (tmp[l]) new_val[l] = true;
			}
			auto itr = dp.find(new_num);
			if (itr != dp.end()) {
				for (int k = 0; k < new_size; k++) if (new_val[k]) itr->second[k] = true;
			} else dp[new_num] = new_val;
		}
	}
	std::vector<int> final(m);
	for (int i = 0; i < m; i++) final[i] = all[i].second;
	assert(dp.count(final));
	auto res = dp[final];
	assert((int) res.size() == x + 1);
	std::string ans;
	for (int i = 1; i <= x; i++) ans.push_back(res[i] ? '1' : '0');
	return ans;
}

int main() {
	int n = ri(), x = ri();
	assert(1 <= n && n <= 100);
	assert(1 <= x && x <= 500000);
	std::vector<int> a(n);
	for (int i = 0; i < n; i++) a[i] = ri(), assert(1 <= a[i] && a[i] <= x);
	clock_t r0 = clock();
	std::cout << solve(n, x, a) << std::endl;
	clock_t r1 = clock();
	std::cerr << (double)(r1 - r0) * 1000 / CLOCKS_PER_SEC << "ms" << std::endl;
	return 0;
}
0