結果

問題 No.563 超高速一人かるた large
ユーザー parukiparuki
提出日時 2017-09-12 11:00:04
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 388 ms / 3,000 ms
コード長 6,372 bytes
コンパイル時間 2,069 ms
コンパイル使用メモリ 179,624 KB
実行使用メモリ 41,472 KB
最終ジャッジ日時 2024-04-25 06:49:25
合計ジャッジ時間 5,239 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 3 ms
5,376 KB
testcase_04 AC 5 ms
5,376 KB
testcase_05 AC 7 ms
6,016 KB
testcase_06 AC 44 ms
14,464 KB
testcase_07 AC 73 ms
18,432 KB
testcase_08 AC 182 ms
27,904 KB
testcase_09 AC 352 ms
40,832 KB
testcase_10 AC 132 ms
32,768 KB
testcase_11 AC 145 ms
35,584 KB
testcase_12 AC 143 ms
35,584 KB
testcase_13 AC 166 ms
35,968 KB
testcase_14 AC 165 ms
35,712 KB
testcase_15 AC 136 ms
35,456 KB
testcase_16 AC 177 ms
35,968 KB
testcase_17 AC 48 ms
15,744 KB
testcase_18 AC 388 ms
41,472 KB
testcase_19 AC 9 ms
10,112 KB
testcase_20 AC 181 ms
27,776 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
#define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i))
#define rep(i,j) FOR(i,0,j)
#define each(x,y) for(auto &(x):(y))
#define mp make_pair
#define MT make_tuple
#define all(x) (x).begin(),(x).end()
#define debug(x) cout<<#x<<": "<<(x)<<endl
#define smax(x,y) (x)=max((x),(y))
#define smin(x,y) (x)=min((x),(y))
#define MEM(x,y) memset((x),(y),sizeof (x))
#define sz(x) (int)(x).size()
#define rt return
using dbl = double;
using ll = long long;
using pii = pair<int, int>;
using vi = vector<int>;
using vll = vector<ll>;

template<int MOD>
class ModInt {
public:
    ModInt() :value(0) {}
    ModInt(long long val) :value((int)(val<0 ? MOD + val%MOD : val%MOD)) { }

    ModInt& operator+=(ModInt that) {
        value = value + that.value;
        if (value >= MOD)value -= MOD;
        return *this;
    }
    ModInt& operator-=(ModInt that) {
        value -= that.value;
        if (value<0)value += MOD;
        return *this;
    }
    ModInt& operator*=(ModInt that) {
        value = (int)((long long)value * that.value % MOD);
        return *this;
    }
    ModInt &operator/=(ModInt that) {
        return *this *= that.inverse();
    }
    ModInt operator+(ModInt that) const {
        return ModInt(*this) += that;
    }
    ModInt operator-(ModInt that) const {
        return ModInt(*this) -= that;
    }
    ModInt operator*(ModInt that) const {
        return ModInt(*this) *= that;
    }
    ModInt operator/(ModInt that) const {
        return ModInt(*this) /= that;
    }
    ModInt pow(long long k) const {
        if (value == 0)return 0;
        ModInt n = *this, res = 1;
        while (k) {
            if (k & 1)res *= n;
            n *= n;
            k >>= 1;
        }
        return res;
    }
    ModInt inverse() const {
        long long a = value, b = MOD, u = 1, v = 0;
        while (b) {
            long long t = a / b;
            a -= t * b;
            swap(a, b);
            u -= t * v;
            swap(u, v);
        }
        return ModInt(u);
    }
    int toi() const { return value; }

private:
    int value;
};
typedef ModInt<1000000007> mint;
ostream& operator<<(ostream& os, const mint& x) {
    os << x.toi();
    return os;
}

struct RollingHash {
    static const int C = 31, C2 = 29, P = 1000000007, P2 = 1000000009;
    int _n;
    vector<long long> pw, pw2, hs, hs2;

    RollingHash() {}
    RollingHash(const string &st)
        :_n((int)st.size()), pw(_n + 1), pw2(_n + 1), hs(_n + 1), hs2(_n + 1)
    {
        pw[0] = pw2[0] = 1;
        rep(i, _n) {
            pw[i + 1] = (pw[i] * C) % P;
            pw2[i + 1] = (pw2[i] * C2) % P2;
            hs[i + 1] = (C*hs[i] + st[i]) % P;
            hs2[i + 1] = (C2*hs2[i] + st[i]) % P2;
        }
    }

    // [l, r)
    pii getHash(int l, int r) {
        int res, res2;
        res = (hs[r] - hs[l] * pw[r - l]) % P;
        res2 = (hs2[r] - hs2[l] * pw2[r - l]) % P2;
        if (res < 0)res += P;
        if (res2 < 0)res2 += P2;
        return mp(res, res2);
    }
};

vector<vector<int>> combinations(int n, int mod) {
    auto res = vector<vector<int>>(n + 1, vector<int>(n + 1));
    rep(i, n + 1) res[i][0] = 1;
    for (int i = 1; i <= n; ++i) for (int j = 1; j <= i; ++j)
        res[i][j] = (res[i - 1][j - 1] + res[i - 1][j]) % mod;
    return res;
}

const int MO = (int)1e9 + 7;
string S[2001];
RollingHash rh[2001];
int f[2001][2001];
mint ans[2001], fact[2001];

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int N;
    cin >> N;
    rep(i, N) {
        cin >> S[i];
    }

    // 接頭辞の問題＝＞ソート
    // これで、接頭辞が似ている文字列ほど近くになる。
    sort(S, S + N);
    rep(i, N) {
        rh[i] = RollingHash(S[i]);
    }

    // f[i][j]はiとjを区別するのに読み上げる文字数
    // ただし、f[i][i] = 1
    rep(i, N)rep(j, N) {
        if (i == j)f[i][j] = 1;
        else if (i > j)f[i][j] = f[j][i];
        else {
            // ローリングハッシュで接頭辞が何文字一致しているか調べる
            int ok = 0, ng = min(sz(S[i]), sz(S[j])) + 1, mid;
            while (ng - ok > 1) {
                mid = (ok + ng) / 2;
                (rh[i].getHash(0, mid) == rh[j].getHash(0, mid) ? ok : ng) = mid;
            }
            f[i][j] = ng;
        }
    }
    // 都合
    f[N - 1][N] = 1;

    auto C = combinations(N + 10, MO);

    /*
    すべてのカードの集合について、読み上げられる文字数は読み札の順番によらない。
    なので読み札の順番をインデックス順にする。
    */
    for (int ba = 1; ba <= N; ++ba) {
        mint sm;
        // 前に取ってないのがa
        // つまり、a+1,a+2,...,a+ba-1はすべて取った状態。
        // a+baを次に取る
        for (int a = 0; a + ba < N; ++a) {
            // a+baから見て
            // 手前はaが一番近い。なので接頭辞が近い。
            // 後ろはまだ見てないので必ず残っている。これは直後が一番近い。
            sm += max(f[a][a + ba], f[a + ba][a + ba + 1]);
        }
        // 全体でK個取る
        // 少なくともaは取ってないのでK<=N-1
        for (int K = ba; K <= N - 1; ++K) {
            // a+1,a+2,...,a+ba-1,a+baは確定している。
            // それとaは使ってはいけない。
            // このとき
            // N-ba-1個の中からK-ba個選ぶ
            if (ba != N)ans[K - 1] += sm * C[N - ba - 1][K - ba];
        }
    }

    // 前にとっていないカードが存在しない
    // つまり、いまのところすべてのカードをとっている場合
    // 0,1,...,b-1まではすべて取った状態
    // bを取るときのコストをしらべる
    for (int b = 0; b < N; ++b) {
        // 全部でK枚とる
        for (int K = b+1; K <= N; ++K) {
            // 直後のカードとだけ比較。前のカードは残っていないので
            // 0からbまで全部取った状態から残りを選ぶ。(C[N-b-1][K-b-1])
            ans[K - 1] += mint(f[b][b + 1]) * C[N - b - 1][K - b - 1];
        }
    }

    // 解（パターンとの積)には並び順も必要。階乗
    fact[0] = 1;
    rep(i, N)fact[i + 1] = fact[i] * (i + 1);
    rep(i, N)cout << ans[i] * fact[i + 1] << endl;
}
0