結果

問題 No.1554 array_and_me
ユーザー 👑 ygussany
提出日時 2021-06-18 21:37:16
言語 C
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,890 bytes
コンパイル時間 1,181 ms
コンパイル使用メモリ 31,744 KB
実行使用メモリ 11,276 KB
最終ジャッジ日時 2024-06-22 20:00:48
合計ジャッジ時間 4,967 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 16 WA * 25
権限があれば一括ダウンロードができます

ソースコード

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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0