結果

問題 No.2531 Coloring Vertices on Namori
コンテスト
ユーザー Nguyễn Phúc Gia Nghi
提出日時 2025-01-29 23:23:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 259 ms / 2,000 ms
コード長 797 bytes
コンパイル時間 1,571 ms
コンパイル使用メモリ 163,652 KB
実行使用メモリ 33,212 KB
最終ジャッジ日時 2025-01-29 23:23:53
合計ジャッジ時間 9,597 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;

int n, f[200010], vis[200010];
ll k, dp[3][200010], ret, c;
const ll mod = 998244353;
vector<int>a[200010];

void dfs(int u, int pre)
{
    vis[u] = 1;
    for (int v : a[u]) if (v != pre)
    {
        if (vis[v]) c = abs(f[v] - f[u]) + 1;
        else f[v] = f[u] + 1, dfs(v, u);
    }
}

int main()
{
    cin >> n >> k;
    for (int i=1; i<=n; i++)
    {
        int x, y;
        cin >> x >> y;
        a[x].push_back(y);
        a[y].push_back(x);
    }
    dfs(1, 0);
    dp[1][1] = k;
    for (int i=2; i<=c; i++)
    {
        dp[1][i] = dp[2][i-1];
        dp[2][i] = (dp[1][i-1] * (k-1) + dp[2][i-1] * (k-2)) % mod;
    }
    ret = dp[2][c];
    for (int i=1; i<=n-c; i++) ret = (ret * (k-1)) % mod;
    cout << ret;
}
0