#387953 (C++14) No.856 増える演算

提出ソース

結果

問題	No.856 増える演算
ユーザー	QCFium
提出日時	2019-10-08 19:01:41
言語	C++14 (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	280 ms / 3,153 ms
コード長	3,127 bytes
コンパイル時間	2,198 ms
コンパイル使用メモリ	179,080 KB
実行使用メモリ	12,968 KB
最終ジャッジ日時	2024-04-25 01:31:14
合計ジャッジ時間	16,759 ms
ジャッジサーバーID （参考情報）	judge4 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	118 ms 10,724 KB
testcase_01	AC	117 ms 10,852 KB
testcase_02	AC	112 ms 10,728 KB
testcase_03	AC	113 ms 10,852 KB
testcase_04	AC	111 ms 10,848 KB
testcase_05	AC	112 ms 10,724 KB
testcase_06	AC	112 ms 10,728 KB
testcase_07	AC	114 ms 10,720 KB
testcase_08	AC	113 ms 10,720 KB
testcase_09	AC	113 ms 10,852 KB
testcase_10	AC	112 ms 10,852 KB
testcase_11	AC	117 ms 10,720 KB
testcase_12	AC	115 ms 10,720 KB
testcase_13	AC	113 ms 10,852 KB
testcase_14	AC	114 ms 10,740 KB
testcase_15	AC	114 ms 10,856 KB
testcase_16	AC	113 ms 10,720 KB
testcase_17	AC	115 ms 10,644 KB
testcase_18	AC	114 ms 10,856 KB
testcase_19	AC	115 ms 10,848 KB
testcase_20	AC	115 ms 10,856 KB
testcase_21	AC	113 ms 10,724 KB
testcase_22	AC	115 ms 10,856 KB
testcase_23	AC	118 ms 11,120 KB
testcase_24	AC	117 ms 10,872 KB
testcase_25	AC	115 ms 10,812 KB
testcase_26	AC	115 ms 10,740 KB
testcase_27	AC	114 ms 10,856 KB
testcase_28	AC	117 ms 10,984 KB
testcase_29	AC	117 ms 11,032 KB
testcase_30	AC	118 ms 11,120 KB
testcase_31	AC	116 ms 10,868 KB
testcase_32	AC	117 ms 11,152 KB
testcase_33	AC	123 ms 11,236 KB
testcase_34	AC	133 ms 11,336 KB
testcase_35	AC	126 ms 11,092 KB
testcase_36	AC	130 ms 11,104 KB
testcase_37	AC	129 ms 11,448 KB
testcase_38	AC	115 ms 10,844 KB
testcase_39	AC	116 ms 10,996 KB
testcase_40	AC	120 ms 10,896 KB
testcase_41	AC	120 ms 11,104 KB
testcase_42	AC	139 ms 11,372 KB
testcase_43	AC	120 ms 10,928 KB
testcase_44	AC	115 ms 10,708 KB
testcase_45	AC	118 ms 10,836 KB
testcase_46	AC	126 ms 10,996 KB
testcase_47	AC	120 ms 10,856 KB
testcase_48	AC	128 ms 11,184 KB
testcase_49	AC	122 ms 11,072 KB
testcase_50	AC	127 ms 11,144 KB
testcase_51	AC	132 ms 11,144 KB
testcase_52	AC	134 ms 11,160 KB
testcase_53	AC	169 ms 11,548 KB
testcase_54	AC	142 ms 11,368 KB
testcase_55	AC	166 ms 11,708 KB
testcase_56	AC	138 ms 11,128 KB
testcase_57	AC	173 ms 11,748 KB
testcase_58	AC	156 ms 11,452 KB
testcase_59	AC	199 ms 12,680 KB
testcase_60	AC	147 ms 11,360 KB
testcase_61	AC	197 ms 11,752 KB
testcase_62	AC	186 ms 12,156 KB
testcase_63	AC	118 ms 10,940 KB
testcase_64	AC	182 ms 11,764 KB
testcase_65	AC	134 ms 11,140 KB
testcase_66	AC	146 ms 11,084 KB
testcase_67	AC	162 ms 11,432 KB
testcase_68	AC	188 ms 12,220 KB
testcase_69	AC	186 ms 12,056 KB
testcase_70	AC	212 ms 12,256 KB
testcase_71	AC	191 ms 12,392 KB
testcase_72	AC	176 ms 11,804 KB
testcase_73	AC	217 ms 12,840 KB
testcase_74	AC	222 ms 12,968 KB
testcase_75	AC	280 ms 12,844 KB
testcase_76	AC	227 ms 12,708 KB
testcase_77	AC	224 ms 12,968 KB
testcase_78	AC	221 ms 12,708 KB
testcase_79	AC	225 ms 12,968 KB
testcase_80	AC	227 ms 12,968 KB
testcase_81	AC	225 ms 12,968 KB
testcase_82	AC	170 ms 12,840 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>

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

int pow(int a, int b, int mod) {
	int res = 1;
	for (; b; b >>= 1) {
		if (b & 1) res = (int64_t) res * a % mod;
		a = (int64_t) a * a % mod;
	}
	return res;
}
int inv(int i, int mod) { return pow(i, mod - 2, mod); }

template<int mod, int proot> struct NTT {
	int get_mod() { return mod; }
	void ntt(std::vector<int> &a, bool inverse) {
		int n = a.size();
		assert((n & -n) == n);
		int h = pow(proot, (mod - 1) / n, mod);
		if (inverse) h = inv(h, mod);
		
		for (int i = 0, j = 1; j < n - 1; j++) {
			for (int k = n >> 1; k > (i ^= k); k >>= 1);
			if (j < i) std::swap(a[i], a[j]);
		}
		for (int i = 1; i < n; i <<= 1) {
			int base = pow(h, n / i / 2, mod);
			int w = 1;
			for (int j = 0; j < i; j++) {
				for (int k = j; k < n; k += i << 1) {
					int u = a[k];
					int d = (int64_t) a[k + i] * w % mod;
					a[k] = u + d;
					if (a[k] >= mod) a[k] -= mod;
					a[k + i] = u - d;
					if (a[k + i] < 0) a[k + i] += mod;
				}
				w = (int64_t) w * base % mod;
			}
		}
		if (inverse) {
			int ninv = inv(a.size(), mod);
			for (auto &i : a) i = (int64_t) i * ninv % mod;
		}
	}
	std::vector<int> conv(const std::vector<int> a_, const std::vector<int> b_) {
		std::vector<int> a = a_, b = b_;
		size_t size = 1;
		for (; size < a_.size() + b_.size(); size <<= 1);
		a.resize(size);
		b.resize(size);
		ntt(a, false);
		ntt(b, false);
		for (size_t i = 0; i < size; i++) a[i] = (int64_t) a[i] * b[i] % mod;
		ntt(a, true);
		a.resize(a_.size() + b_.size() - 1);
		return a;
	}
};
#define MOD 1000000007

std::vector<int64_t> conv(const std::vector<int> a, const std::vector<int> b) {
	NTT<998244353, 3> ntt1;
	NTT<469762049, 3> ntt2;
	auto r0 = ntt1.conv(a, b);
	auto r1 = ntt2.conv(a, b);
	int n = r0.size();
	std::vector<int64_t> res;
	for (int i = 0; i < n; i++) {
		int64_t cur = r0[i];
		while (cur % 469762049 != r1[i]) cur += 998244353;
		res.push_back(cur);
	}
	return res;
}
int main() {
	int n = ri();
	int a[n];
	for (int i = 0; i < n; i++) a[i] = ri();
	
	std::multiset<std::pair<int, int> > all;
	for (int i = 0; i < n; i++) all.insert({a[i], i});
	std::pair<int, int> optimal;
	double min = 1000000000;
	for (int i = 0; i + 1 < n; i++) {
		all.erase(all.find({a[i], i}));
		double cur = log2(a[i]) * all.begin()->first + log2(a[i] + all.begin()->first);
		if (min > cur) min = cur, optimal = {i, all.begin()->second};
	}
	
	int64_t sum[n];
	sum[n - 1] = 0;
	for (int i = n - 2; i >= 0; i--) sum[i] = sum[i + 1] + a[i + 1];
	int res = 1;
	for (int i = 0; i < n; i++) res = (int64_t) res * pow(a[i], sum[i] % (MOD - 1), MOD) % MOD;
	
	std::vector<int> num(100001);
	for (int i = 0; i < n; i++) num[a[i]]++;
	auto conved = conv(num, num);
	for (int i = 0; i < n; i++) conved[a[i] + a[i]]--;
	for (int i = 1; i <= 200000; i++) assert(conved[i] % 2 == 0), conved[i] >>= 1, res = (int64_t) res * pow(i, conved[i] % (MOD - 1), MOD) % MOD;
	res = (int64_t) res * inv((int64_t) (a[optimal.first] + a[optimal.second]) * pow(a[optimal.first], a[optimal.second], MOD) % MOD, MOD) % MOD;
	
	std::cout << res << std::endl;
	
	return 0;
}

yukicoder

結果

テストケース

ソースコード