結果

問題 No.444 旨味の相乗効果
コンテスト
ユーザー pekempey
提出日時 2016-12-27 21:42:02
言語 C++14
(gcc 15.2.0 + boost 1.90.0)
コンパイル:
g++-15 -O2 -lm -std=c++14 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 2 ms / 2,500 ms
コード長 1,125 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,628 ms
コンパイル使用メモリ 202,704 KB
実行使用メモリ 9,228 KB
最終ジャッジ日時 2026-07-06 04:57:46
合計ジャッジ時間 3,430 ms
ジャッジサーバーID
（参考情報） 		judge2_1 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 23
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:62:23: warning: 'k' may be used uninitialized [-Wmaybe-uninitialized]
   62 |                 for (k++; k < dp.size(); k++) {
      |                      ~^~
main.cpp:57:21: note: 'k' was declared here
   57 |                 int k;
      |                     ^

ソースコード

diff #
raw source code 

#pragma GCC optimize ("O3")
#pragma GCC target ("avx")
#include <bits/stdc++.h>
using namespace std;

const int mod = 1e9 + 7;

int add(int a, int b) {
	return (a += b) >= mod ? a - mod : a;
}

int mul(int a, int b) {
	return 1LL * a * b % mod;
}

void upd(int &a, int b) {
	if ((a += b) >= mod) {
		a -= mod;
	}
}

int modpow(int a, long long b) {
	int ret = 1;
	while (b > 0) {
		if (b & 1) {
			ret = 1LL * ret * a % mod;
		}
		a = 1LL * a * a % mod;
		b >>= 1;
	}
	return ret;
}

int main() {
	int n;
	long long c;
	cin >> n >> c;
	
	int foo = 0;
	vector<int> a(n);
	for (int i = 0; i < n; i++) {
		cin >> a[i];
		upd(foo, mod - modpow(a[i], c));
	}

	vector<int> dp(n * 2);

	dp[0] = 1;

	// doubling
	for (int ii = 0; ii < 61; ii++) {
		for (int i = 0; i < n; i++) {
			for (int j = dp.size() - 2; j >= 0; j--) {
				upd(dp[j + 1], mul(dp[j], a[i]));
			}
		}
		int k;
		for (int j = (c & 1); j < dp.size(); j += 2) {
			k = j / 2;
			dp[k] = dp[j];
		}
		for (k++; k < dp.size(); k++) {
			dp[k] = 0;
		}
		for (int i = 0; i < n; i++) {
			a[i] = mul(a[i], a[i]);
		}
		c >>= 1;
	}

	cout << add(dp[0], foo) << endl;
}
0