結果

問題 No.1554 array_and_me
ユーザー 👑 ygussany
提出日時 2021-06-18 21:49:35
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 101 ms / 2,000 ms
コード長 2,386 bytes
コンパイル時間 642 ms
コンパイル使用メモリ 31,872 KB
実行使用メモリ 9,600 KB
最終ジャッジ日時 2024-06-22 20:11:23
合計ジャッジ時間 4,918 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <stdio.h>
const int Mod = 998244353;
long long fact[100001], fact_inv[100001];
typedef struct {
    long double key;
    int id;
} data;
typedef struct {
    data obj[100001];
    int size;
} min_heap;
 
void push(data x, min_heap* h)
{
    int i = ++(h->size), j = i >> 1;
    data tmp;
    h->obj[i] = x;
    while (j > 0) {
        if (h->obj[i].key < h->obj[j].key) {
            tmp = h->obj[j];
            h->obj[j] = h->obj[i];
            h->obj[i] = tmp;
            i = j;
            j >>= 1;
        } else break;
    }
}
data pop(min_heap* h)
{
    int i = 1, j = 2;
    data output = h->obj[1], tmp;
    h->obj[1] = h->obj[(h->size)--];
    while (j <= h->size) {
        if (j < h->size && h->obj[j^1].key < h->obj[j].key) j ^= 1;
        if (h->obj[j].key < h->obj[i].key) {
            tmp = h->obj[j];
            h->obj[j] = h->obj[i];
            h->obj[i] = tmp;
            i = j;
            j = i << 1;
        } else break;
    }
    return output;
}
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, j, k = 0, N, K, sum = 0;
    long long ans = 1;
    data d;
    static int A[100001], num[100001];
    static long long prob[100001];
    static long double likely[100001];
    static min_heap h;
    h.size = 0;
    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);
        likely[i] = (long double)A[i] * K / sum;
        num[i] = (long long)A[i] * K / sum;
        k += num[i];
        d.key = (num[i] + 1 - likely[i]) / likely[i];
        d.id = i;
        push(d, &h);
    }
    for (i = 1; i <= K - k; i++) {
        d = pop(&h);
        j = d.id;
        num[j]++;
        d.key = (num[j] + 1 - likely[j]) / likely[j];
        push(d, &h);
    }
    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