QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#698395#7895. Graph Partitioning 2test_algthRE 34ms15924kbC++142.2kb2024-11-01 19:19:542024-11-01 19:19:55

Judging History

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

  • [2024-11-01 19:19:55]
  • 评测
  • 测评结果:RE
  • 用时:34ms
  • 内存:15924kb
  • [2024-11-01 19:19:54]
  • 提交

answer

#include <bits/stdc++.h>

const int MAXN = 1.01E5;
const int MAXB = 500;
const int P = 998244353;

inline int add(int a, int b) {
  return a -= (a += b) >= P ? P : 0;
}

inline void incr(int& a, int b) {
  a = add(a, b);
}

int dp[MAXN][MAXB + 5], sz[MAXN], ndp[MAXB + 5];
int N, K;
std::unordered_map <int, int> f[MAXN], g;
std::vector <int> adj[MAXN];

inline void merge(int u, int v) {
  for (int i = 0; i <= std::min(K + 1, sz[u] + sz[v]); ++i)
    ndp[i] = 0;
  
  for (int i = 0; i <= sz[u]; ++i) {
    for (int j = 0; j <= sz[v] && i + j <= K + 1; ++j) {
      incr(ndp[i + j], 1ll * dp[u][i] * dp[v][j] % P);
    }
    incr(ndp[i], 1ll * dp[u][i] * add(dp[v][K], dp[v][K + 1]) % P);
  }

  sz[u] += sz[v];
  for (int i = 0; i <= std::min(K + 1, sz[u]); ++i)
    dp[u][i] = ndp[i];
}

void solve(int u, int pa) {
  sz[u] = dp[u][1] = 1;
  for (int i = 2; i <= K + 1; ++i)
    dp[u][i] = 0;
  for (int v : adj[u]) {
    if (v == pa) continue;
    solve(v, u);
    merge(u, v);
  }
}

inline void merge2(int u, int v) {
  g.clear();
  for (auto [a, b] : f[u]) {
    for (auto [c, d] : f[v]) {
      if (a + c <= K + 1) {
        incr(g[a + c], 1ll * b * d % P);
      }
    }
    
    if (f[v].count(K)) {
      incr(g[a], 1ll * b * f[v][K] % P);
    }

    if (f[v].count(K + 1)) {
      incr(g[a], 1ll * b * f[v][K + 1]);
    }
  }

  f[u] = g;
}

void solve2(int u, int pa) {
  f[u].clear();
  f[u][1] = 1;
  for (int v : adj[u]) {
    if (v == pa) continue;
    solve2(v, u);
    merge2(u, v);
  }
}

int main() {
  std::ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
  std::cout.tie(nullptr);

  int testCase = 1;
  std::cin >> testCase;
  while (testCase--) {
    std::cin >> N >> K;
    for (int i = 1; i <= N; ++i) {
      adj[i].clear();
    }

    for (int i = 1, u, v; i < N; ++i) {
      std::cin >> u >> v;
      adj[u].push_back(v);
      adj[v].push_back(u);
    }

    if (K < MAXB) { 
      solve(1, 1);
      std::cout << add(dp[1][K], dp[1][K + 1]) << '\n';
    }
    else {
      solve2(1, 1);
      std::cout << add(f[1][K], f[1][K + 1]) << '\n';
    }
  }

  return 0;
}

詳細信息

Test #1:

score: 100
Accepted
time: 4ms
memory: 14104kb

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: 34ms
memory: 15924kb

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
Runtime Error

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: