QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#750837#7895. Graph Partitioning 2WedonotLikeStudying#WA 31ms10076kbC++201.4kb2024-11-15 16:06:042024-11-15 16:06:09

Judging History

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

  • [2024-11-15 16:06:09]
  • 评测
  • 测评结果:WA
  • 用时:31ms
  • 内存:10076kb
  • [2024-11-15 16:06:04]
  • 提交

answer

#include<bits/stdc++.h>
#define rep(i,j,k) for (int i=(j);i<=(k);++i)
#define dep(i,j,k) for (int i=(j);i>=(k);--i)
#define MAXN 100000
#define MOD 998244353
#define fi first
#define se second
using namespace std;

vector<int> g[MAXN+5];
unordered_map<int,int> dp[MAXN+5],tmp[2];
int k;

void dfs(int x,int fa) {
    for (auto i:g[x]) {
        if (i==fa) continue;
        dfs(i,x);
    }
    tmp[0].clear();tmp[1].clear();
    tmp[0].insert(make_pair(1,1));
    int t=0;
    for (auto i:g[x]) {
        if (i==fa) continue;
        tmp[t^1].clear();
        for (auto j:dp[i]) {
            for (auto l:tmp[t]) {
                if (j.fi+l.fi<=k+1) {
                    tmp[t^1][j.fi+l.fi]=(tmp[t^1][j.fi+l.fi]+1ll*j.se*l.se)%MOD;
                }
            }
        }
        t^=1;
    }
    for (auto i:tmp[t]) {
        if (i.fi==k) {
            dp[x][k]=i.se;
            dp[x][0]+=i.se;
        } else if (i.fi==k+1) {
            dp[x][0]+=i.se;
        } else {
            dp[x][i.fi]=i.se;
        }
    }
}

void solve() {
    int n,u,v;
    cin>>n>>k;
    rep(i,1,n) g[i].clear();
    rep(i,1,n) dp[i].clear();
    rep(i,1,n-1) {
        cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(1,0);
    cout<<dp[1][0]<<"\n";
    return;
}

int main() {
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int t;
    cin>>t;
    while (t--) solve();
    return 0;
}

详细

Test #1:

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

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: -100
Wrong Answer
time: 31ms
memory: 10076kb

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:

wrong answer 5004th lines differ - expected: '224400650', found: '1222645003'