結果

問題 No.2531 Coloring Vertices on Namori
ユーザー 👑 AngrySadEightAngrySadEight
提出日時 2023-10-19 02:46:06
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 455 ms / 2,000 ms
コード長 1,883 bytes
コンパイル時間 3,599 ms
コンパイル使用メモリ 176,720 KB
実行使用メモリ 44,760 KB
最終ジャッジ日時 2024-09-18 22:45:51
合計ジャッジ時間 14,856 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 2 ms
5,376 KB
testcase_05 AC 261 ms
43,008 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 262 ms
43,136 KB
testcase_08 AC 413 ms
43,008 KB
testcase_09 AC 455 ms
43,136 KB
testcase_10 AC 435 ms
43,008 KB
testcase_11 AC 234 ms
44,760 KB
testcase_12 AC 234 ms
44,760 KB
testcase_13 AC 234 ms
44,636 KB
testcase_14 AC 406 ms
43,008 KB
testcase_15 AC 417 ms
43,008 KB
testcase_16 AC 391 ms
43,008 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 410 ms
44,032 KB
testcase_21 AC 410 ms
44,032 KB
testcase_22 AC 408 ms
44,160 KB
testcase_23 AC 415 ms
44,032 KB
testcase_24 AC 404 ms
44,160 KB
testcase_25 AC 399 ms
44,032 KB
testcase_26 AC 405 ms
44,032 KB
testcase_27 AC 407 ms
44,160 KB
testcase_28 AC 413 ms
44,032 KB
testcase_29 AC 413 ms
44,032 KB
testcase_30 AC 417 ms
44,032 KB
testcase_31 AC 418 ms
44,160 KB
testcase_32 AC 391 ms
44,032 KB
testcase_33 AC 404 ms
44,032 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/all>
#include <cassert>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <utility>
#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]--;
    }
    map<pair<ll, ll>, ll> mp;
    for (ll i = 0; i < N; i++) {
        mp[pair<ll, ll>(u[i], v[i])]++;
    }
    for (ll i = 0; i < N; i++) {
        ll val = mp[pair<ll, ll>(u[i], v[i])];
        assert(val == 1);
    }
    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