QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#945041#9805. Guide MapElevenXWA 1ms3584kbC++201.2kb2025-03-20 18:47:172025-03-20 18:47:20

Judging History

This is the latest submission verdict.

  • [2025-03-20 18:47:20]
  • Judged
  • Verdict: WA
  • Time: 1ms
  • Memory: 3584kb
  • [2025-03-20 18:47:17]
  • Submitted

answer

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

const int MOD = 998244353;

vector<vector<int>> adj;
vector<bool> visited;
int s = 0;

void dfs(int u) {
    visited[u] = true;
    s++;
    for (int v : adj[u]) {
        if (!visited[v]) {
            dfs(v);
        }
    }
}

long long pow_mod(long long a, long long b) {
    long long res = 1;
    while (b) {
        if (b & 1) res = res * a % MOD;
        a = a * a % MOD;
        b >>= 1;
    }
    return res;
}

int main() {
    int n;
    cin >> n;
    if (n == 2) {
        cout << 2 << endl;
        return 0;
    }
    adj.resize(n + 1);
    for (int i = 0; i < n-2; i++) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    visited.assign(n + 1, false);
    dfs(1);
    int t = n - s;

    long long bridge = (long long)s * t;
    long long within_s = (long long)s * (s - 1) / 2;
    long long within_t = (long long)t * (t - 1) / 2;
    long long m = ((long long)(s-1)*(s-2)/2) + ((long long)(t-1)*(t-2)/2);

    long long ans = (bridge + within_s + within_t) % MOD;
    ans = ans * pow_mod(2, m) % MOD;

    cout << ans << endl;
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3584kb

input:

4
1 4
2 3

output:

6

result:

ok 1 number(s): "6"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3584kb

input:

2

output:

2

result:

ok 1 number(s): "2"

Test #3:

score: -100
Wrong Answer
time: 0ms
memory: 3456kb

input:

4
1 2
1 3

output:

12

result:

wrong answer 1st numbers differ - expected: '6', found: '12'