QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#395833#7895. Graph Partitioning 2GenshinImpactsFault#TL 51ms11936kbC++201.2kb2024-04-21 20:29:242024-04-21 20:29:24

Judging History

你现在查看的是最新测评结果

  • [2024-04-21 20:29:24]
  • 评测
  • 测评结果:TL
  • 用时:51ms
  • 内存:11936kb
  • [2024-04-21 20:29:24]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;
#define MOD 998244353
map<int, int> f[100010];
vector<int> G[100010];
pair<int, int> g[100010];
int n, K;
void dfs(int x, int fa) {
    for(auto i : G[x]) {
        if(i == fa) continue;
        dfs(i, x);
    }
    map<int, int> mp;
    f[x][1] = 1;
    for(auto i : G[x]) {
        if(i == fa) continue;
        mp.clear();
        for(auto [j, k] : f[i]) {
            for(auto [j1, k1] : f[x]) {
                (mp[j + j1] += 1ll * k * k1 % MOD) %= MOD;
                if(j == K || j == K + 1) {
                    (mp[j1] += 1ll * k * k1 % MOD) %= MOD;
                }
            }
        }
        f[x] = mp;
    }
}
void Do() {
    cin >> n >> K;
    for(int i = 1; i <= n; ++ i) {
        f[i].clear();
        G[i].clear();
    }
    for(int i = 1; i <= n; ++ i) {
        g[i].first = g[i].second = 0;
    }
    for(int i = 1; i < n; ++ i) {
        int u, v;
        cin >> u >> v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1, 0);
    int Ans = (f[1][K] + f[1][K + 1]) % MOD;
    cout << Ans << "\n";
}
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    int T; cin >> T;
    while(T --) {
        Do();
    }
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 2ms
memory: 9828kb

input:

2
8 2
1 2
3 1
4 6
3 5
2 4
8 5
5 7
4 3
1 2
1 3
2 4

output:

2
1

result:

ok 2 lines

Test #2:

score: 0
Accepted
time: 51ms
memory: 11936kb

input:

5550
13 4
10 3
9 1
10 8
3 11
8 5
10 7
9 6
13 5
9 7
2 7
5 12
4 8
8 2
4 1
3 4
7 8
2 5
6 7
4 8
2 3
11 1
11 10
1 4
9 10
8 4
3 6
5 7
6 1
10 2
11 7
11 1
17 2
14 16
13 15
17 3
15 11
1 6
13 2
13 17
4 8
14 10
8 14
14 5
9 12
14 2
12 17
17 6
15 7
14 6
2 14
2 13
2 4
8 4
3 11
7 3
14 1
11 9
13 3
5 10
6 8
3 10
14 ...

output:

0
3
112
0
1
0
1
0
0
0
1
0
1
0
0
1
0
140
0
0
0
814
1
6
1
1
2
2
0
612
0
1
0
0
0
1
1
0
0
121
4536
0
0
1718
0
0
1
0
444
1
1908
1813
3
74
0
1
0
46
0
0
0
0
0
0
0
0
0
1
0
1
1
1
239
0
0
0
1
0
0
0
1
0
1
0
0
1
1
0
0
0
1
0
0
0
48
0
2
0
0
0
1
364
0
206
0
0
76
0
1
0
0
2
0
1
2
0
0
1
0
0
4
0
1
1
0
0
1
1
1
0
0
1
1
...

result:

ok 5550 lines

Test #3:

score: -100
Time Limit Exceeded

input:

3
99990 259
23374 69108
82204 51691
8142 67119
48537 97966
51333 44408
33147 68485
21698 86824
15746 58746
78761 86975
58449 61819
69001 68714
25787 2257
25378 14067
64899 68906
29853 31359
75920 85420
76072 11728
63836 55505
43671 98920
77281 25176
40936 66517
61029 61440
66908 52300
92101 59742
69...

output:


result: