結果

問題 No.856 増える演算
ユーザー kakira9618kakira9618
提出日時 2019-03-28 12:19:10
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 5,219 bytes
コンパイル時間 2,101 ms
コンパイル使用メモリ 176,868 KB
実行使用メモリ 30,396 KB
最終ジャッジ日時 2024-04-17 22:19:28
合計ジャッジ時間 28,982 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 289 ms
27,784 KB
testcase_01 AC 290 ms
27,908 KB
testcase_02 AC 282 ms
27,720 KB
testcase_03 AC 289 ms
27,808 KB
testcase_04 AC 293 ms
27,712 KB
testcase_05 AC 289 ms
27,820 KB
testcase_06 AC 287 ms
27,704 KB
testcase_07 AC 294 ms
27,744 KB
testcase_08 AC 294 ms
27,784 KB
testcase_09 AC 290 ms
27,908 KB
testcase_10 AC 291 ms
27,860 KB
testcase_11 AC 292 ms
27,712 KB
testcase_12 AC 288 ms
27,724 KB
testcase_13 AC 290 ms
27,716 KB
testcase_14 AC 289 ms
27,916 KB
testcase_15 AC 298 ms
27,956 KB
testcase_16 AC 291 ms
27,748 KB
testcase_17 AC 293 ms
27,860 KB
testcase_18 AC 288 ms
27,768 KB
testcase_19 AC 295 ms
27,780 KB
testcase_20 AC 296 ms
27,848 KB
testcase_21 AC 290 ms
27,812 KB
testcase_22 AC 284 ms
27,856 KB
testcase_23 AC 292 ms
28,004 KB
testcase_24 AC 295 ms
28,040 KB
testcase_25 AC 293 ms
27,812 KB
testcase_26 AC 291 ms
27,964 KB
testcase_27 AC 294 ms
27,764 KB
testcase_28 AC 293 ms
28,068 KB
testcase_29 AC 292 ms
27,880 KB
testcase_30 AC 291 ms
27,864 KB
testcase_31 AC 282 ms
27,844 KB
testcase_32 AC 287 ms
28,036 KB
testcase_33 AC 292 ms
28,144 KB
testcase_34 AC 295 ms
28,420 KB
testcase_35 AC 290 ms
28,168 KB
testcase_36 AC 298 ms
28,560 KB
testcase_37 AC 296 ms
28,496 KB
testcase_38 AC 290 ms
27,800 KB
testcase_39 AC 293 ms
27,888 KB
testcase_40 AC 296 ms
28,088 KB
testcase_41 AC 294 ms
28,012 KB
testcase_42 AC 298 ms
28,496 KB
testcase_43 AC 293 ms
28,060 KB
testcase_44 AC 288 ms
27,716 KB
testcase_45 AC 289 ms
27,948 KB
testcase_46 AC 294 ms
28,016 KB
testcase_47 AC 292 ms
27,988 KB
testcase_48 AC 291 ms
28,216 KB
testcase_49 AC 293 ms
28,088 KB
testcase_50 AC 293 ms
28,336 KB
testcase_51 AC 294 ms
28,416 KB
testcase_52 AC 293 ms
28,468 KB
testcase_53 WA -
testcase_54 WA -
testcase_55 WA -
testcase_56 AC 307 ms
28,180 KB
testcase_57 WA -
testcase_58 WA -
testcase_59 WA -
testcase_60 WA -
testcase_61 WA -
testcase_62 WA -
testcase_63 AC 290 ms
27,900 KB
testcase_64 WA -
testcase_65 AC 304 ms
28,100 KB
testcase_66 WA -
testcase_67 WA -
testcase_68 WA -
testcase_69 WA -
testcase_70 WA -
testcase_71 WA -
testcase_72 WA -
testcase_73 WA -
testcase_74 WA -
testcase_75 WA -
testcase_76 WA -
testcase_77 WA -
testcase_78 WA -
testcase_79 WA -
testcase_80 WA -
testcase_81 WA -
testcase_82 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

// O((N + max(A_i))log max(A_i))解
// A_i > 50000 以上の場合に起こりうるバグを修正したバージョン

#include <bits/stdc++.h>

using namespace std;
#define all(x) (x).begin(),(x).end()
#define rep(i, n) for (int i = 0; i < (n); i++)
#define chmin(x, y) (x) = min((x), (y))
#define chmax(x, y) (x) = max((x), (y))
#define endl "\n"
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef complex<double> C;

template <typename T> ostream &operator<<(ostream &os, const vector<T> &vec) {os << "["; for (const auto &v : vec) {os << v << ","; } os << "]"; return os;}
template <typename T, typename U> ostream &operator<<(ostream &os, const pair<T, U> &p) {os << "(" << p.first << ", " << p.second << ")"; return os;}

const int mod = 1e9 + 7;
const int MAX_A = 100000;

ll mod_pow(ll x, ll n, ll mod) {
    ll res = 1;
    while(n > 0) {
        if(n & 1) res = res * x % mod;
            x = x * x % mod;
            n >>= 1;
    }
    return res;
}

ll inv(ll x) {
    return mod_pow(x, mod - 2, mod);
}

ll f(ll x, ll y) {
    return (((x + y) % mod) * mod_pow(x, y, mod)) % mod;
}

// (x + y)x^y < (s + t)s^t
// log(x + y) + ylog(x) < log(s + t) + tlog(s)
double g(ll x, ll y) {
    return log(x + y) + y * log(x);
}


 
void fft(vector<C> &f, int n, int dir) {
    if (n == 1) return;
    vector<C> f0, f1;
    for(int i = 0; i < n; i++) {
        if (i % 2 == 0) {
            f0.push_back(f[i]);
        } else {
            f1.push_back(f[i]);
        }
    }
    fft(f0, n / 2, dir);
    fft(f1, n / 2, dir);
    C zeta = polar(1.0, 2 * M_PI * dir / n);
    C now = C(1.0, 0.0);
    for(int i = 0; i < n; i++) {
        f[i] = f0[i % (n / 2)] + now * f1[i % (n / 2)];
        now *= zeta;
    }
}
 
vector<C> comb(vector<C> A, vector<C> B) {
    int size = A.size() + B.size();
    int n = 1;
    while(n < size) n <<= 1;
    A.resize(n);
    B.resize(n);
    fft(A, n, 1);
    fft(B, n, 1);
    vector<C> ret;
    for(int i = 0; i < n; i++) {
        ret.push_back(A[i] * B[i]);
    }
    fft(ret, n, -1);
    for(int i = 0; i < n; i++) {
        ret[i] /= n;
    }
    return ret;
}

vector<ll> getPairCnt(vector<ll> &K) {
    vector<C> A, B;
    for (int i = 0; i < K.size(); i++) {
        A.push_back(C(K[i], 0));
        B.push_back(C(K[i], 0));
    }
    vector<ll> ret;
    vector<C> ret_ = comb(A, B);
    for (int i = 0; i < ret_.size(); i++) {
        ret.push_back((ll)(real(ret_[i]) + 0.1));
    }
    for (int i = 0; i < ret.size(); i++) {
        if (i % 2 == 0) ret[i] -= (i / 2 < K.size() ? K[i / 2] : 0);
        ret[i] >>= 1;
    }
    return ret;
}

void solve() {
    int N;
    cin >> N;
    assert(N >= 1 && N <= MAX_A);
    vector<ll> A(N), K(MAX_A + 1);
    for (int i = 0; i < N; i++) {
        cin >> A[i];
        assert(A[i] >= 1 && A[i] <= MAX_A);
        A[i] %= mod;
        K[A[i]]++;
    }

    // 考察1. miをO(N)で求める
    // iを固定すると，jとしては（i<jの中で）もちろんA[j]が小さいのを選ぶべき．
    // i以降の要素で最も小さい要素をO(1)で求められるようなテーブルをO(N)かけて前処理する．

    vector<ll> mi_table(N + 1, 1LL << 60);
    for (int i = N - 1; i >= 0; i--) {
        mi_table[i] = min(mi_table[i + 1], A[i]);
    }

    double mi = g(A[0], mi_table[1]);
    ll mi_f = f(A[0], mi_table[1]);
    for (int i = 1; i < N; i++) {
        if (mi > g(A[i], mi_table[i + 1])) {
            mi = g(A[i], mi_table[i + 1]);
            mi_f = f(A[i], mi_table[i + 1]);
        }
    }

    // 考察2. \prod_i \prod_j f(A[i], A[j]) をO(N)で求める
    // 考察2a.
    // 塁乗部分は各iについて A[i]^{\sum_{i<j}{A[j]}} を計算すれば良いので
    // \sum_{i<j}{A[j]} を前計算しておけばできる

    vector<ll> acc_back(N + 1);
    for (int i = N - 1; i >= 0; i--) {
        acc_back[i] = acc_back[i + 1] + A[i];
        acc_back[i] %= mod;
    }
    
    ll prod = 1;
    for (int i = 0; i < N; i++) {
        prod *= mod_pow(A[i], acc_back[i + 1], mod);
        prod %= mod;
    }

    // 考察2b.
    // あとはA[i] + A[j] = k (i < j)となるような(i, j)が何個あるかを解けば良い
    // A[i]が小さいので，kを決め打ちして探索できる．

    // vector<ll> pair_cnt(2001);
    // for (int k = 2; k <= 2000; k++) {
    //     ll c = 0;
    //     for(int a = 0; a <= k; a++) {
    //         int b = k - a;
    //         c += K[a] * K[b];
    //         c %= mod;
    //     }
    //     if (k % 2 == 0) c -= K[k / 2];
    //     pair_cnt[k] += c / 2;
    // }

    // 考察2b-2.
    // 実は上の処理はFFTによる畳込み演算を用いて高速に計算できる．
    // kが偶数の場合にペア数を少し補正する必要があることに注意．
    vector<ll> pair_cnt = getPairCnt(K);

    for (int k = 0; k < pair_cnt.size(); k++) {
        prod *= mod_pow(k, pair_cnt[k], mod);
        prod %= mod;
    }

    cout << (prod * inv(mi_f)) % mod << endl;
    
}


int main() {
    #ifdef LOCAL_ENV
    cin.exceptions(ios::failbit);
    #endif
    cin.tie(0);
    ios::sync_with_stdio(false);
    cout.setf(ios::fixed);
    cout.precision(16);
    
    solve();
}
0