QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#253141#7608. CliquesShayan86TL 3ms13992kbC++143.1kb2023-11-16 18:45:272023-11-16 18:45:27

Judging History

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

  • [2023-11-16 18:45:27]
  • 评测
  • 测评结果:TL
  • 用时:3ms
  • 内存:13992kb
  • [2023-11-16 18:45:27]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;

#pragma GCC optimize("O3,unroll-loops")
// #pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
// Ofast, O0, O1, O2, O3, unroll-loops, fast-math, trapv

typedef long long ll;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;

#define Mp          make_pair
#define sep         ' '
#define endl        '\n'
#define F	        first
#define S	        second
#define pb          push_back
#define all(x)      (x).begin(),(x).end()
#define kill(res)	cout << res << '\n', exit(0);
#define set_dec(x)	cout << fixed << setprecision(x);
#define fast_io     ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define file_io     freopen("input.txt", "r", stdin) ; freopen("output.txt", "w", stdout);

#define lid (id*2)
#define rid (id*2+1)
#define mid ((l+r)/2)

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

const ll N = 2e5 + 50;
const ll Mod = 1e9 + 7;

ll n, q, par[N], sz[N], head[N], st[N], timer = 1, p2[N];

vector<int> adj[N];

ll mod(ll a, ll b = Mod){
    while(a >= b) a -= b;
    return a;
}

void pre(int v){
    if(par[v]) adj[v].erase(find(all(adj[v]), par[v]));
    sz[v] = 1;
    for(int u: adj[v]){
        par[u] = v; pre(u); sz[v] += sz[u];
    }
    for(int i = 1; i < adj[v].size(); i++) if(sz[adj[v][0]] < sz[adj[v][i]]) swap(adj[v][0], adj[v][i]);
}

void hld(int v){
    st[v] = timer++;
    for(int u: adj[v]){
        if(u == adj[v][0]) head[u] = head[v];
        else head[u] = u;
        hld(u);
    }
}

int bit1[N], bit2[N];

void upd1(int x, int y){
    for(; x <= n; x += x & (-x)) bit1[x] += y;
}

int get1(int x){
    int out = 0;
    for(; x; x &= x-1) out += bit1[x];
    return out;
}

void upd2(int x, int y){
    for(; x <= n; x += x & (-x)) bit2[x] += y;
}

int get2(int x){
    int out = 0;
    for(; x; x &= x-1) out += bit2[x];
    return out;
}

void upd_path1(int u, int v, int x){
    while(head[u] != head[v]){
        if(st[u] < st[v]) swap(u, v);
        upd1(st[head[u]], x); upd1(st[u]+1, -x);
        u = par[head[u]];
    }
    if(st[u] < st[v]) swap(u, v);
    upd1(st[v], x); upd1(st[u]+1, -x);
}

void upd_path2(int u, int v, int x){
    while(head[u] != head[v]){
        if(st[u] < st[v]) swap(u, v);
        upd2(st[head[u]], x); upd2(st[u]+1, -x);
        u = par[head[u]];
    }
    if(st[u] < st[v]) swap(u, v);
    upd2(st[v]+1, x); upd2(st[u]+1, -x);
}

int main(){
    fast_io;

    p2[0] = 1;
    for(int i = 1; i < N; i++) p2[i] = mod(p2[i-1] + p2[i-1]);

    cin >> n;

    int u, v;
    for(int i = 1; i < n; i++){
        cin >> u >> v;
        adj[u].pb(v);
        adj[v].pb(u);
    }

    pre(1); head[1] = 1; hld(1);

    cin >> q;

    while(q--){
        char t;
        cin >> t >> u >> v;

        if(t == '+'){
            upd_path1(u, v, 1); upd_path2(u, v, 1);
        }
        else{
            upd_path1(u, v, -1); upd_path2(u, v, -1);
        }

        ll res = 0;
        for(int i = 1; i <= n; i++) res = mod(res + p2[get1(i)] - p2[get2(i)] + Mod);

        cout << res << endl;
    }

}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

5
1 2
5 1
2 3
4 2
6
+ 4 5
+ 2 2
+ 1 3
- 2 2
+ 2 3
+ 4 4

output:

1
3
7
3
7
9

result:

ok 6 lines

Test #2:

score: 0
Accepted
time: 3ms
memory: 13652kb

input:

20
8 7
19 10
12 14
3 16
17 13
7 10
5 6
1 9
15 12
5 2
16 13
3 11
20 14
18 6
1 14
16 20
11 10
3 4
20 6
30
+ 10 20
+ 14 20
+ 12 17
- 14 20
- 12 17
+ 4 6
+ 8 20
+ 3 6
- 10 20
+ 2 17
+ 1 16
+ 6 10
+ 9 10
+ 5 15
+ 7 8
- 7 8
+ 2 5
+ 3 18
+ 1 20
+ 8 16
- 3 18
- 5 15
+ 4 20
+ 14 16
- 4 6
+ 8 19
+ 4 7
- 1 16
...

output:

1
3
7
3
1
3
7
15
7
15
31
63
127
255
257
255
259
515
1027
1283
643
385
769
1537
769
785
865
481
737
369

result:

ok 30 lines

Test #3:

score: -100
Time Limit Exceeded

input:

131072
3641 72511
10338 18718
8949 15478
79108 62887
42154 28540
65359 102645
93744 48493
3277 103211
43542 61315
93315 118634
24021 57937
31034 436
30411 6208
1388 25159
93424 128520
119820 87281
5860 97559
59648 8225
57244 58766
119685 13716
130165 60958
79806 116338
97486 80167
101963 95499
51263...

output:

1
2
4
8
13
27
43
67
119
215
359
423
847
1167
1935
2447
2975
4511
5023
8639
6911
13759
12991
15103
23295
43775
47999
91647
85375
167295
169343
301695
600703
666239
1332351
2664575
2678911
2687103
5374207
5898495
11731455
11731967
22283263
22807551
21955327
43910399
43911423
44207359
44207360
44469504...

result: