結果

問題 No.2531 Coloring Vertices on Namori
ユーザー AngrySadEightAngrySadEight
提出日時 2023-10-19 01:28:19
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 264 ms / 2,000 ms
コード長 1,629 bytes
コンパイル時間 4,622 ms
コンパイル使用メモリ 168,352 KB
実行使用メモリ 32,456 KB
最終ジャッジ日時 2023-10-19 01:28:32
合計ジャッジ時間 11,981 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 2 ms
4,348 KB
testcase_04 AC 2 ms
4,348 KB
testcase_05 AC 166 ms
30,692 KB
testcase_06 AC 2 ms
4,348 KB
testcase_07 AC 167 ms
30,692 KB
testcase_08 AC 230 ms
30,692 KB
testcase_09 AC 225 ms
30,692 KB
testcase_10 AC 251 ms
30,692 KB
testcase_11 AC 141 ms
32,456 KB
testcase_12 AC 141 ms
32,456 KB
testcase_13 AC 141 ms
32,456 KB
testcase_14 AC 217 ms
30,692 KB
testcase_15 AC 215 ms
30,692 KB
testcase_16 AC 222 ms
30,692 KB
testcase_17 AC 2 ms
4,348 KB
testcase_18 AC 2 ms
4,348 KB
testcase_19 AC 2 ms
4,348 KB
testcase_20 AC 233 ms
31,748 KB
testcase_21 AC 236 ms
31,748 KB
testcase_22 AC 238 ms
31,748 KB
testcase_23 AC 236 ms
31,748 KB
testcase_24 AC 254 ms
31,748 KB
testcase_25 AC 235 ms
31,748 KB
testcase_26 AC 236 ms
31,748 KB
testcase_27 AC 242 ms
31,748 KB
testcase_28 AC 240 ms
31,748 KB
testcase_29 AC 264 ms
31,748 KB
testcase_30 AC 236 ms
31,748 KB
testcase_31 AC 236 ms
31,748 KB
testcase_32 AC 237 ms
31,748 KB
testcase_33 AC 240 ms
31,748 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/all>
#include <cassert>
#include <iostream>
#include <queue>
#include <set>
#include <vector>
using namespace std;
using namespace atcoder;
using ll = long long;

ll mod = 998244353;


int main() {
    ll N, K;
    cin >> N >> K;
    vector<ll> u(N);
    vector<ll> v(N);
    for (ll i = 0; i < N; i++) {
        cin >> u[i] >> v[i];
        u[i]--;
        v[i]--;
    }
    vector<ll> dist(N, N + 2);
    queue<ll> que;
    vector<vector<ll>> graph(N, vector<ll>(0));
    dsu tree(N);
    ll cnt = 0;
    for (ll i = 0; i < N; i++) {
        if (!tree.same(u[i], v[i])) {
            graph[u[i]].push_back(v[i]);
            graph[v[i]].push_back(u[i]);
            tree.merge(u[i], v[i]);
        } else {
            que.push(u[i]);
            dist[u[i]] = 0;
            while (que.size()) {
                ll p = que.front();
                que.pop();
                for (ll x : graph[p]) {
                    if (dist[x] < N) {
                        continue;
                    }
                    dist[x] = dist[p] + 1;
                    que.push(x);
                }
            }
            cnt = dist[v[i]] + 1;
            break;
        }
    }
    vector<vector<ll>> dp(N, vector<ll>(2, 0));
    dp[0][0] = K;
    for (ll i = 1; i < N; i++) {
        dp[i][0] = dp[i - 1][1];
        dp[i][1] = (dp[i - 1][0] * (K - 1) + dp[i - 1][1] * (K - 2)) % mod;
    }
    ll ans = dp[cnt - 1][1];
    for (ll i = 0; i < N - cnt; i++) {
        ans = (ans * (K - 1)) % mod;
    }
    cout << ans << endl;
    for (int i = 0; i < N; i++) {
        assert(tree.same(0, i));
    }
    return 0;
}
0