結果

問題 No.2531 Coloring Vertices on Namori
ユーザー AAPPeeAAPPee
提出日時 2023-11-22 19:39:17
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 211 ms / 2,000 ms
コード長 1,629 bytes
コンパイル時間 3,008 ms
コンパイル使用メモリ 152,020 KB
実行使用メモリ 32,260 KB
最終ジャッジ日時 2024-09-26 07:38:58
合計ジャッジ時間 9,261 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 156 ms
30,700 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 156 ms
30,596 KB
testcase_08 AC 189 ms
30,636 KB
testcase_09 AC 179 ms
30,696 KB
testcase_10 AC 180 ms
30,720 KB
testcase_11 AC 131 ms
32,260 KB
testcase_12 AC 128 ms
32,236 KB
testcase_13 AC 130 ms
32,160 KB
testcase_14 AC 174 ms
30,716 KB
testcase_15 AC 178 ms
30,720 KB
testcase_16 AC 181 ms
30,696 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 208 ms
31,488 KB
testcase_21 AC 197 ms
31,580 KB
testcase_22 AC 202 ms
31,488 KB
testcase_23 AC 199 ms
31,528 KB
testcase_24 AC 200 ms
31,616 KB
testcase_25 AC 199 ms
31,488 KB
testcase_26 AC 211 ms
31,588 KB
testcase_27 AC 202 ms
31,476 KB
testcase_28 AC 192 ms
31,616 KB
testcase_29 AC 194 ms
31,488 KB
testcase_30 AC 193 ms
31,616 KB
testcase_31 AC 192 ms
31,604 KB
testcase_32 AC 193 ms
31,488 KB
testcase_33 AC 195 ms
31,500 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