QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#301209#7895. Graph Partitioning 2luyiming123TL 50ms13024kbC++141.4kb2024-01-09 16:02:292024-01-09 16:02:30

Judging History

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

  • [2024-01-09 16:02:30]
  • 评测
  • 测评结果:TL
  • 用时:50ms
  • 内存:13024kb
  • [2024-01-09 16:02:29]
  • 提交

answer

# include <bits/stdc++.h>
using namespace std;
using i64 = long long; 
const int N = 1e5 + 5; 
const i64 mod = 998244353; 
void Add(i64 &x, i64 y) {x = (x + y) % mod; if(x < 0) x += mod; return; }
unordered_map<int, i64> dp[N]; 
vector <int> g[N]; 
int n, k, siz[N]; 
void dfs(int x, int fa)
{
    dp[x][1] = 1, siz[x] = 1; 
    for(int v : g[x])
    {
        if(v == fa) continue; 
        dfs(v, x); 
        unordered_map<int, i64> tmp; tmp.clear(); 
        for(auto it = dp[x].begin(); it != dp[x].end(); it++)
        {
            for(auto itv = dp[v].begin(); itv != dp[v].end(); itv++)
            {
                Add(tmp[(it -> first + itv -> first)], it -> second * itv -> second % mod);
                if(itv -> first == k || itv -> first == k + 1) Add(tmp[it -> first], itv -> second * it -> second % mod); 
            }
        }
        dp[x] = tmp; 
    }
    return; 
} 
int main(void)
{
    int Test; scanf("%d", &Test);
    while(Test--)
    {
        scanf("%d%d", &n, &k); 
        for(int i = 1; i <= n; i++) 
        {
            g[i].clear(), dp[i].clear();
        }
        for(int i = 1; i < n; i++)
        {
            int u, v; scanf("%d%d", &u, &v); 
            g[u].push_back(v), g[v].push_back(u); 
        }
        dfs(1, 0); 
        printf("%lld\n", (dp[1][k] + dp[1][k + 1]) % mod); 
    }
    
    return 0; 
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 11532kb

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: 50ms
memory: 13024kb

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: