QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#288029#7961. 一棵树raymond_7AC ✓914ms112460kbC++143.3kb2023-12-21 16:17:482023-12-21 16:17:48

Judging History

This is the latest submission verdict.

  • [2023-12-21 16:17:48]
  • Judged
  • Verdict: AC
  • Time: 914ms
  • Memory: 112460kb
  • [2023-12-21 16:17:48]
  • Submitted

answer

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <vector>

#define PII pair<int,int>
#define x first
#define y second
#define For(i, l, r) for(int i = l; i <= r; i ++)

void in(int &x)
{
    char c = getchar();
    while(c > '9' || c < '0') c = getchar();
    for(x = 0; c >= '0' && c <= '9'; c = getchar())
        x = x * 10 + c - '0';
}

using namespace std;

const int N = 5e5 + 50, M = N * 21;

int n, m;
vector<int> G[N];
int g[N], siz[N], son[N];
int rt[N], tot;
signed lc[M], rc[M], sz[M], tag[M], val[M], key[M];

int New(int v)
{
    tot ++; lc[tot] = rc[tot] = 0; tag[tot] = 0; sz[tot] = 1; val[tot] = v; key[tot] = rand(); return tot;
}

int Rt;

void Add(int u, int v)
{
    tag[u] += v; val[u] += v;
}

void down(int u)
{
    if(!tag[u] || !u) return ;
    if(lc[u]) Add(lc[u], tag[u]);
    if(rc[u]) Add(rc[u], tag[u]);
    tag[u] = 0;
}

void debug(int u)
{
    if(!u) return ; down(u);
    debug(lc[u]); g[++ g[0]] = val[u]; debug(rc[u]);
}

void up(int u) {sz[u] = sz[lc[u]] + sz[rc[u]] + 1;}

int Rank(int u, int x)
{
    if(!u) return 0; down(u);
    if(x <= val[u]) return Rank(lc[u], x);
    return Rank(rc[u], x) + 1 + sz[lc[u]];
}

void split(int u, int &x, int &y, int k)
{
    // if(!k) {x = 0; y = u; return ;}
    if(!u) {x = y = 0; return ;}
    down(u);
    if(k <= sz[lc[u]]) y = u, split(lc[u], x, lc[y], k);
    else x = u, split(rc[u], rc[x], y, k - 1 - sz[lc[u]]);
    up(u);
}

void merge(int &u, int x, int y)
{
    if(!x || !y) {u = x + y; return ;}
    down(x); down(y);
    if(key[x] > key[y]) u = x, merge(rc[u], rc[x], y);
    else u = y, merge(lc[u], x, lc[y]);
    up(u);
}

void ins(int &rt, int v)
{
    int rk = Rank(rt, v), x, y, z = New(v);
    split(rt, x, y, rk); merge(y, z, y); merge(rt, x, y);
}

void Dfs(int u, int pa)
{
    siz[u] = 1;
    if(G[u].size() == 1 && u > 1)
    {
        rt[u] = New(abs(m - 2) - m);
        return ;
    }
    For(ii, 0, (int)G[u].size() - 1)
    // for(int v : G[u])
    {
        int v = G[u][ii];
        if(v == pa) continue;
        Dfs(v, u);
        if(siz[v] > siz[son[u]]) son[u] = v;
        siz[u] += siz[v];
    }
    rt[u] = rt[son[u]];
    ins(rt[u], 0);
    For(i, 0, (int)G[u].size() - 1)
    {
        int v = G[u][i];
        if(v == pa || v == son[u]) continue;
        Rt = rt[u];
        g[0] = 0;
        debug(rt[v]);
        For(j, 1, g[0]) ins(rt[u], g[j]);
     }
    if(u == 1) return ;
    if(siz[u] <= m / 2)
    {
        Add(rt[u], -2);
    }
    else 
    {
        int x, y, z;
        if(m & 1)
        {
            split(rt[u], x, y, m / 2);
            Add(x, -2); split(y, z, y, 1);
            Add(y, 2); merge(y, z, y);
            merge(rt[u], x, y);
        }
        else
        {
            split(rt[u], x, y, m / 2);
            Add(x, -2); Add(y, 2);
            merge(rt[u], x, y);
        }
    }
}

signed main()
{
    // freopen("pub.in", "r", stdin);
    // freopen("pub.out", "w", stdout);
    in(n); in(m);
    For(i, 1, n - 1) 
    {
        int u, v; in(u); in(v);
        G[u].push_back(v); G[v].push_back(u);
    }
    Dfs(1, 0);
    long long ans = 1ll * (n - 1) * m;
    g[0]=  0; debug(rt[1]);
    For(i, 1, m) ans += g[i];
    printf("%lld\n", ans);
    return 0;
}

詳細信息

Test #1:

score: 100
Accepted
time: 860ms
memory: 104276kb

input:

499991 31473
1 2
2 3
1 4
2 5
4 6
4 7
6 8
4 9
8 10
6 11
9 12
10 13
10 14
1 15
14 16
3 17
14 18
12 19
13 20
11 21
16 22
16 23
20 24
5 25
16 26
18 27
8 28
15 29
27 30
27 31
26 32
21 33
3 34
27 35
33 36
25 37
22 38
11 39
11 40
17 41
25 42
3 43
3 44
2 45
35 46
25 47
5 48
6 49
41 50
42 51
1 52
39 53
14 54...

output:

15734930984

result:

ok single line: '15734930984'

Test #2:

score: 0
Accepted
time: 330ms
memory: 78232kb

input:

499993 461706
1 2
2 3
3 4
1 5
3 6
3 7
7 8
5 9
5 10
7 11
10 12
9 13
13 14
12 15
13 16
15 17
14 18
17 19
16 20
15 21
16 22
17 23
22 24
20 25
23 26
25 27
27 28
28 29
25 30
29 31
30 32
31 33
33 34
33 35
32 36
36 37
35 38
33 39
34 40
35 41
41 42
40 43
42 44
39 45
40 46
41 47
43 48
44 49
46 50
49 51
50 52...

output:

195357165512

result:

ok single line: '195357165512'

Test #3:

score: 0
Accepted
time: 345ms
memory: 78168kb

input:

499993 170472
1 2
1 3
1 4
4 5
5 6
4 7
6 8
5 9
8 10
6 11
6 12
10 13
13 14
11 15
14 16
15 17
14 18
16 19
16 20
17 21
17 22
20 23
19 24
21 25
23 26
22 27
27 28
24 29
24 30
28 31
31 32
30 33
29 34
29 35
35 36
33 37
34 38
36 39
36 40
36 41
41 42
40 43
41 44
39 45
45 46
42 47
47 48
44 49
45 50
46 51
50 52...

output:

65062419628

result:

ok single line: '65062419628'

Test #4:

score: 0
Accepted
time: 289ms
memory: 78204kb

input:

499994 801
1 2
2 3
1 4
2 5
2 6
1 7
7 8
4 9
6 10
10 11
7 12
7 13
10 14
13 15
15 16
14 17
16 18
18 19
15 20
17 21
18 22
19 23
21 24
23 25
21 26
26 27
27 28
24 29
28 30
29 31
27 32
30 33
28 34
29 35
35 36
31 37
36 38
36 39
38 40
36 41
36 42
38 43
38 44
42 45
41 46
41 47
47 48
46 49
45 50
49 51
49 52
48...

output:

285905307

result:

ok single line: '285905307'

Test #5:

score: 0
Accepted
time: 298ms
memory: 78232kb

input:

499990 275
1 2
2 3
2 4
3 5
3 6
4 7
5 8
4 9
4 10
10 11
7 12
9 13
10 14
12 15
13 16
15 17
16 18
18 19
16 20
19 21
20 22
17 23
23 24
22 25
20 26
23 27
22 28
26 29
27 30
27 31
29 32
27 33
33 34
33 35
31 36
31 37
32 38
37 39
35 40
37 41
39 42
37 43
38 44
43 45
42 46
43 47
43 48
47 49
45 50
49 51
47 52
47...

output:

98388719

result:

ok single line: '98388719'

Test #6:

score: 0
Accepted
time: 335ms
memory: 78232kb

input:

499999 419261
1 2
2 3
3 4
2 5
3 6
4 7
2 8
6 9
8 10
10 11
10 12
11 13
13 14
12 15
10 16
16 17
14 18
16 19
17 20
16 21
17 22
17 23
20 24
20 25
22 26
26 27
26 28
27 29
24 30
25 31
27 32
30 33
32 34
29 35
34 36
31 37
37 38
38 39
34 40
40 41
37 42
42 43
43 44
40 45
45 46
42 47
42 48
46 49
46 50
45 51
47 ...

output:

174917393258

result:

ok single line: '174917393258'

Test #7:

score: 0
Accepted
time: 317ms
memory: 78184kb

input:

499999 839
1 2
1 3
3 4
2 5
2 6
4 7
6 8
6 9
7 10
10 11
11 12
10 13
12 14
11 15
14 16
13 17
12 18
18 19
16 20
15 21
16 22
17 23
19 24
24 25
24 26
26 27
23 28
24 29
26 30
28 31
29 32
30 33
28 34
34 35
35 36
31 37
35 38
34 39
36 40
35 41
39 42
42 43
40 44
44 45
40 46
41 47
43 48
44 49
47 50
50 51
47 52
...

output:

299869724

result:

ok single line: '299869724'

Test #8:

score: 0
Accepted
time: 350ms
memory: 112460kb

input:

499994 469174
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
10 11
11 12
12 13
13 14
14 15
15 16
16 17
17 18
18 19
19 20
20 21
21 22
22 23
23 24
24 25
25 26
26 27
27 28
28 29
29 30
30 31
31 32
32 33
33 34
34 35
35 36
36 37
37 38
38 39
39 40
40 41
41 42
42 43
43 44
44 45
45 46
46 47
47 48
48 49
49 50
50 51
51 ...

output:

110061651964

result:

ok single line: '110061651964'

Test #9:

score: 0
Accepted
time: 468ms
memory: 112456kb

input:

499994 688
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
10 11
11 12
12 13
13 14
14 15
15 16
16 17
17 18
18 19
19 20
20 21
21 22
22 23
23 24
24 25
25 26
26 27
27 28
28 29
29 30
30 31
31 32
32 33
33 34
34 35
35 36
36 37
37 38
38 39
39 40
40 41
41 42
42 43
43 44
44 45
45 46
46 47
47 48
48 49
49 50
50 51
51 52
...

output:

235984

result:

ok single line: '235984'

Test #10:

score: 0
Accepted
time: 359ms
memory: 112364kb

input:

499990 141264
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
10 11
11 12
12 13
13 14
14 15
15 16
16 17
17 18
18 19
19 20
20 21
21 22
22 23
23 24
24 25
25 26
26 27
27 28
28 29
29 30
30 31
31 32
32 33
33 34
34 35
35 36
36 37
37 38
38 39
39 40
40 41
41 42
42 43
43 44
44 45
45 46
46 47
47 48
48 49
49 50
50 51
51 ...

output:

9977617584

result:

ok single line: '9977617584'

Test #11:

score: 0
Accepted
time: 627ms
memory: 112408kb

input:

499999 210371
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
10 11
11 12
12 13
13 14
14 15
15 16
16 17
17 18
18 19
19 20
20 21
21 22
22 23
23 24
24 25
25 26
26 27
27 28
28 29
29 30
30 31
31 32
32 33
33 34
34 35
35 36
36 37
37 38
38 39
39 40
40 41
41 42
42 43
43 44
44 45
45 46
46 47
47 48
48 49
49 50
50 51
51 ...

output:

22128058078

result:

ok single line: '22128058078'

Test #12:

score: 0
Accepted
time: 868ms
memory: 104972kb

input:

499997 427984
1 2
2 3
1 4
2 5
5 6
5 7
5 8
1 9
3 10
3 11
5 12
7 13
3 14
14 15
10 16
8 17
6 18
1 19
11 20
14 21
9 22
6 23
5 24
8 25
1 26
17 27
20 28
27 29
19 30
18 31
22 32
18 33
4 34
16 35
14 36
6 37
17 38
36 39
33 40
37 41
27 42
21 43
32 44
18 45
27 46
33 47
35 48
2 49
47 50
32 51
48 52
40 53
11 54
...

output:

213979261810

result:

ok single line: '213979261810'

Test #13:

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

input:

1 1

output:

0

result:

ok single line: '0'

Test #14:

score: 0
Accepted
time: 914ms
memory: 107044kb

input:

499990 449066
1 2
2 3
1 4
2 5
4 6
4 7
7 8
6 9
8 10
9 11
5 12
3 13
3 14
3 15
13 16
7 17
13 18
11 19
11 20
18 21
5 22
18 23
10 24
8 25
5 26
22 27
24 28
18 29
5 30
23 31
12 32
19 33
20 34
34 35
16 36
5 37
24 38
10 39
25 40
16 41
18 42
28 43
12 44
28 45
14 46
15 47
34 48
16 49
40 50
3 51
30 52
40 53
8 5...

output:

224515847140

result:

ok single line: '224515847140'

Test #15:

score: 0
Accepted
time: 823ms
memory: 105020kb

input:

499998 296976
1 2
1 3
3 4
1 5
2 6
4 7
7 8
8 9
7 10
1 11
11 12
6 13
4 14
4 15
10 16
2 17
9 18
10 19
1 20
7 21
19 22
7 23
18 24
17 25
7 26
19 27
11 28
12 29
25 30
25 31
9 32
8 33
12 34
22 35
31 36
13 37
12 38
27 39
5 40
29 41
18 42
37 43
25 44
25 45
38 46
4 47
1 48
16 49
34 50
6 51
6 52
22 53
47 54
25...

output:

148478316284

result:

ok single line: '148478316284'

Test #16:

score: 0
Accepted
time: 879ms
memory: 107240kb

input:

499991 620
1 2
2 3
2 4
3 5
3 6
1 7
2 8
5 9
7 10
8 11
11 12
1 13
12 14
10 15
11 16
12 17
10 18
9 19
12 20
17 21
21 22
6 23
17 24
23 25
16 26
10 27
23 28
13 29
11 30
26 31
18 32
6 33
15 34
3 35
32 36
32 37
34 38
17 39
37 40
26 41
24 42
40 43
21 44
34 45
5 46
4 47
21 48
12 49
5 50
11 51
39 52
7 53
31 5...

output:

309963582

result:

ok single line: '309963582'

Test #17:

score: 0
Accepted
time: 826ms
memory: 104484kb

input:

499996 264554
1 2
1 3
3 4
4 5
4 6
4 7
2 8
1 9
8 10
7 11
7 12
1 13
10 14
2 15
3 16
14 17
4 18
5 19
6 20
16 21
8 22
19 23
22 24
20 25
7 26
4 27
26 28
24 29
8 30
15 31
15 32
14 33
25 34
17 35
32 36
1 37
6 38
5 39
11 40
5 41
20 42
21 43
36 44
29 45
32 46
20 47
10 48
29 49
35 50
33 51
48 52
32 53
26 54
3...

output:

132267626666

result:

ok single line: '132267626666'

Test #18:

score: 0
Accepted
time: 277ms
memory: 78168kb

input:

499997 252
1 2
1 3
2 4
2 5
4 6
6 7
6 8
8 9
4 10
7 11
8 12
11 13
8 14
12 15
12 16
11 17
15 18
15 19
16 20
18 21
20 22
21 23
21 24
20 25
23 26
21 27
24 28
26 29
27 30
26 31
31 32
28 33
29 34
29 35
30 36
31 37
35 38
37 39
37 40
35 41
36 42
39 43
39 44
42 45
41 46
44 47
42 48
46 49
46 50
47 51
48 52
47 ...

output:

89992070

result:

ok single line: '89992070'

Test #19:

score: 0
Accepted
time: 308ms
memory: 78240kb

input:

499997 935
1 2
1 3
1 4
2 5
1 6
3 7
3 8
7 9
8 10
10 11
10 12
10 13
9 14
12 15
15 16
16 17
15 18
13 19
14 20
18 21
18 22
17 23
21 24
21 25
21 26
24 27
27 28
25 29
29 30
27 31
29 32
27 33
32 34
33 35
35 36
34 37
37 38
35 39
37 40
37 41
40 42
40 43
43 44
41 45
42 46
43 47
46 48
48 49
47 50
50 51
50 52
4...

output:

334159450

result:

ok single line: '334159450'

Test #20:

score: 0
Accepted
time: 323ms
memory: 78244kb

input:

499994 876
1 2
1 3
1 4
1 5
1 6
5 7
3 8
7 9
6 10
7 11
7 12
8 13
13 14
11 15
12 16
13 17
14 18
16 19
19 20
19 21
19 22
18 23
21 24
19 25
21 26
24 27
26 28
27 29
28 30
28 31
29 32
29 33
33 34
33 35
31 36
33 37
35 38
34 39
36 40
38 41
40 42
41 43
38 44
43 45
42 46
42 47
47 48
45 49
44 50
47 51
51 52
48 ...

output:

312803574

result:

ok single line: '312803574'