結果

問題 No.1554 array_and_me
ユーザー 👑 ygussanyygussany
提出日時 2021-06-18 21:37:16
言語 C
(gcc 12.3.0)
結果
WA  
実行時間 -
コード長 1,890 bytes
コンパイル時間 1,188 ms
コンパイル使用メモリ 30,988 KB
実行使用メモリ 11,308 KB
最終ジャッジ日時 2023-09-04 22:33:49
合計ジャッジ時間 5,655 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
7,424 KB
testcase_01 AC 32 ms
7,520 KB
testcase_02 AC 32 ms
7,508 KB
testcase_03 AC 34 ms
7,580 KB
testcase_04 AC 33 ms
7,600 KB
testcase_05 AC 33 ms
7,500 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 AC 76 ms
11,220 KB
testcase_12 AC 79 ms
11,104 KB
testcase_13 AC 87 ms
11,160 KB
testcase_14 AC 85 ms
11,196 KB
testcase_15 AC 95 ms
11,168 KB
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 AC 75 ms
7,804 KB
testcase_37 AC 73 ms
7,820 KB
testcase_38 AC 76 ms
7,812 KB
testcase_39 AC 79 ms
7,776 KB
testcase_40 AC 72 ms
7,704 KB
testcase_41 AC 34 ms
7,492 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>

const int Mod = 998244353;
long long fact[100001], fact_inv[100001];

typedef struct {
	long double key;
	int id;
} data;

void merge_sort(data x[], int n)
{
	static data y[100001] = {};
	if (n <= 1) return;
	merge_sort(&(x[0]), n/2);
	merge_sort(&(x[n/2]), (n+1)/2);
	
	int i, p, q;
	for (i = 0, p = 0, q = n/2; i < n; i++) {
		if (p >= n/2) y[i] = x[q++];
		else if (q >= n) y[i] = x[p++];
		else y[i] = (x[p].key < x[q].key)? x[p++]: x[q++];
	}
	for (i = 0; i < n; i++) x[i] = y[i];
}

long long div_mod(long long x, long long y, long long z)
{
	if (x % y == 0) return x / y;
	else return (div_mod((1 + x / y) * y - x, (z % y), y) * z + x) / y;
}

long long pow_mod(int n, int k)
{
	long long N, ans = 1;
	for (N = n; k > 0; k >>= 1, N = N * N % Mod) if (k & 1) ans = ans * N % Mod;
	return ans;
}

long long combination(int n, int k)
{
	return fact[n] * fact_inv[k] % Mod * fact_inv[n-k] % Mod;
}

void solve()
{
	int i, k = 0, N, K, sum = 0;
	long long ans = 1;
	static int A[100001], num[100001];
	static long long prob[100001];
	static data d[100001];
	scanf("%d %d", &N, &K);
	for (i = 0; i < N; i++) {
		scanf("%d", &(A[i]));
		sum += A[i];
	}
	for (i = 0; i < N; i++) {
		prob[i] = div_mod(A[i], sum, Mod);
		num[i] = (long long)A[i] * K / sum;
		k += num[i];
		d[i].key = (long double)A[i] * K / sum - num[i];
		d[i].id = i;
	}
	merge_sort(d, N);
	for (i = 1; i <= K - k; i++) num[d[N-i].id]++;
	for (i = 0; i < N; i++) {
		ans = ans * pow_mod(prob[i], num[i]) % Mod * combination(K, num[i]) % Mod;
		K -= num[i];
	}
	printf("%lld\n", ans);
}

int main()
{
	int i;
	for (i = 1, fact[0] = 1; i <= 100000; i++) fact[i] = fact[i-1] * i % Mod;
	for (i = 99999, fact_inv[100000] = div_mod(1, fact[100000], Mod); i >= 0; i--) fact_inv[i] = fact_inv[i+1] * (i + 1) % Mod;

	int t, T;
	scanf("%d", &T);
	for (i = 1; i <= T; i++) solve();
	fflush(stdout);
	return 0;
}
0