結果

問題 No.1510 Simple Integral
ユーザー maine_honzuki
提出日時 2021-05-15 00:01:22
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,618 bytes
コンパイル時間 4,719 ms
コンパイル使用メモリ 263,096 KB
最終ジャッジ日時 2025-01-21 12:41:43
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#include "atcoder/all"
using namespace atcoder;
using Mint = modint998244353;

int main() {
    int N;
    cin >> N;
    vector<int> A(N);
    map<int, int> mp;
    for (auto&& a : A) {
        cin >> a;
        mp[a]++;
    }
    vector<pair<int, int>> V;
    for (auto p : mp) {
        V.push_back(p);
    }
    Mint ans = 0;
    for (auto&& [a, n] : V) {
        Mint book = 1;
        vector<Mint> memo(n);
        memo[0] = 1;
        for (auto&& b : A) {
            if (b != a) {
                {
                    vector<Mint> memo_nxt(n);
                    for (int i = 0; i <= n - 1; i++) {
                        Mint tmp = Mint{1} / (a + b);
                        for (int j = 0;; j++) {
                            if (i + j > n - 1)
                                break;
                            //j回微分
                            memo_nxt[i + j] += memo[i] * tmp;
                            tmp *= -(j + 1);
                            tmp /= (a + b);
                        }
                    }
                    swap(memo, memo_nxt);
                }
                {
                    vector<Mint> memo_nxt(n);
                    for (int i = 0; i <= n - 1; i++) {
                        Mint tmp = Mint{1} / (a - b);
                        for (int j = 0;; j++) {
                            if (i + j > n - 1)
                                break;
                            //j回微分
                            memo_nxt[i + j] += memo[i] * tmp;
                            tmp *= -(j + 1);
                            tmp /= (a - b);
                        }
                    }
                    swap(memo, memo_nxt);
                }
            } else {
                {
                    vector<Mint> memo_nxt(n);
                    for (int i = 0; i <= n - 1; i++) {
                        Mint tmp = Mint{1} / (a + a);
                        for (int j = 0;; j++) {
                            if (i + j > n - 1)
                                break;
                            //j回微分
                            memo_nxt[i + j] += memo[i] * tmp;
                            tmp *= -(j + 1);
                            tmp /= (a + a);
                        }
                    }
                    swap(memo, memo_nxt);
                }
            }
        }
        book = memo[n - 1];
        for (int i = 1; i <= n - 1; i++) {
            book /= i;
        }
        ans += book;
    }
    if (N % 2 == 0)
        ans *= -1;
    ans *= 2;
    cout << ans.val() << endl;
}
0