結果

問題 No.1621 Sequence Inversions
ユーザー だれだれ
提出日時 2021-07-12 22:57:10
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 134 ms / 3,000 ms
コード長 1,425 bytes
コンパイル時間 930 ms
コンパイル使用メモリ 87,492 KB
実行使用メモリ 104,960 KB
最終ジャッジ日時 2024-07-17 15:55:28
合計ジャッジ時間 5,124 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 61 ms
103,092 KB
testcase_01 AC 59 ms
103,168 KB
testcase_02 AC 64 ms
103,468 KB
testcase_03 AC 60 ms
103,168 KB
testcase_04 AC 109 ms
103,296 KB
testcase_05 AC 64 ms
103,168 KB
testcase_06 AC 61 ms
103,160 KB
testcase_07 AC 65 ms
103,144 KB
testcase_08 AC 106 ms
103,296 KB
testcase_09 AC 88 ms
103,208 KB
testcase_10 AC 87 ms
103,168 KB
testcase_11 AC 106 ms
103,316 KB
testcase_12 AC 122 ms
103,296 KB
testcase_13 AC 126 ms
103,336 KB
testcase_14 AC 131 ms
103,868 KB
testcase_15 AC 131 ms
104,956 KB
testcase_16 AC 119 ms
104,832 KB
testcase_17 AC 130 ms
104,880 KB
testcase_18 AC 121 ms
104,812 KB
testcase_19 AC 64 ms
103,436 KB
testcase_20 AC 95 ms
104,192 KB
testcase_21 AC 134 ms
104,960 KB
testcase_22 AC 132 ms
104,832 KB
testcase_23 AC 133 ms
104,852 KB
testcase_24 AC 62 ms
103,352 KB
testcase_25 AC 60 ms
103,424 KB
testcase_26 AC 61 ms
103,216 KB
testcase_27 AC 60 ms
103,280 KB
testcase_28 AC 61 ms
103,168 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <iostream>
#include <vector>
#include <atcoder/modint>

using namespace std;
using mint = atcoder::modint998244353;

mint f[101][101][2501];

int main(){
    for (int i = 0; i < 101; i++){
        f[0][i][0] = 1;
    }
    for (int n = 1; n < 101; n++){
        for (int k = 0; k < 101; k++){
            for (int m = 0; m < 2501; m++){
                if (n * k < m){
                    break;
                }
                f[n][k][m] += f[n - 1][k][m];
                if (m - n >= 0 && k > 0){
                    f[n][k][m] += f[n][k - 1][m - n];
                }
            }
        }
    }
    int t, k;
    cin >> t >> k;
    vector<int> a(t);
    for (int i = 0; i < t; i++){
        cin >> a[i];
    }
    sort(a.begin(), a.end());

    vector<int> cnt(100);
    int n = 0;
    for (int i = 0; i < t - 1; i++){
        cnt[n]++;
        if (a[i] != a[i + 1]){
            n++;
        }
    }
    cnt[n++]++;

    vector dp(n + 1, vector<mint>(5000, 0));
    dp[0][0] = 1;
    int cursum = 0;

    for (int i = 1; i <= n; i++){
        auto x = cnt[i - 1];
        for (int j = 0; j < 5000; j++){
            for (int l = 0; l <= cursum * x; l++){
                if (j - l < 0){
                    break;
                }
                dp[i][j] += f[x][cursum][l] * dp[i - 1][j - l];
            }
        }
        cursum += x;
    }

    cout << dp[n][k].val() << endl;
}
0