oj.uz

Submission #478124

# Submission time Handle Problem Language Result Execution time Memory
478124 2021-10-05T17:42:53 Z bonopo Dynamic Diameter (CEOI19_diameter) C++17
49 / 100
 5000 ms 226304 KB 
#include "bits/stdc++.h"
using namespace std;
#define pb push_back
#define rc (i<<1)|1
#define lc (i<<1)
#define el "\n"
#define f first
#define s second
 
typedef long long ll;
const int MM=1e5+5, MOD=1e9+7, LOG=19;
int N, Q, W, sz[MM], bk[MM], pa[LOG][MM], in[LOG][MM], ot[LOG][MM], cp[LOG][MM], tp[LOG][MM], ct;
vector<int> adj[MM]; ll av[MM];
multiset<ll> ast;
unordered_map<int,int> mp[MM];
 
struct node {
    ll lz;
    pair<ll,int> mx;
};
 
struct tree {
    vector<node> seg;
    int wt;
 
    void prop(int i, int l, int r) {
        if(l!=r) {
            seg[lc].lz+=seg[i].lz;
            seg[rc].lz+=seg[i].lz; }
        seg[i].mx.f+=seg[i].lz;
        seg[i].lz=0;
    }
 
    void build(int i, int sl, int sr) {
        if(sl==sr) {
            seg[i].mx={0,sl};
            return; }
        int m=sl+sr>>1;
        build(rc, m+1, sr);
        build(lc, sl, m);
        calc(i);
    }
 
    void calc(int i) {
        seg[i].mx=max(seg[lc].mx, seg[rc].mx);
    }
 
    void upd(int i, int sl, int sr, int l, int r, ll v) {
        prop(i, sl, sr);
        if(l>sr || r<sl) return;
        if(l<=sl&&sr<=r) {
            seg[i].lz+=v;
            prop(i, sl, sr); return; }
        int m=sl+sr>>1;
        upd(rc, m+1, sr, l, r, v);
        upd(lc, sl, m, l, r, v);
        calc(i);
    }
 
    pair<ll,int> rmq(int i, int sl, int sr, int l, int r) {
        prop(i, sl, sr);
        if(l>sr || r<sl) return {0,0};
        if(l<=sl&&sr<=r) return seg[i].mx;
        int m=sl+sr>>1;
        return max(rmq(lc, sl, m, l, r), rmq(rc, m+1, sr, l, r));
    }
 
    void init(int n) {
        wt=n;
        seg.resize(4*n+4);
        build(1, 1, n);
    }
} st[MM];
 
int crt(int u, int th, int fa) {
    for(int &v:adj[u]) if(v!=fa&&!bk[v]&&sz[v]>=th) {
        return crt(v, th, u); }
    return u;
}
 
void dfs(int u, int fa) {
    sz[u]=1;
    for(int &v:adj[u]) if(v!=fa&&!bk[v]) {
        dfs(v, u);
        sz[u]+=sz[v];
    }
}
 
void etr(int u, int fa, int lvl, int rt, int tv) {
    in[lvl][u]=++ct; cp[lvl][u]=rt; tp[lvl][u]=tv; mp[rt][ct]=u;
    for(int &v:adj[u]) if(v!=fa&&!bk[v]) {
        etr(v, u, lvl, rt, tv); }
    ot[lvl][u]=ct;
}
 
void rec(int rt, int nds, int fa=-1, int lvl=0) {
    dfs(rt, 0);
 
    // find centroid
  //  cout<<"centroid for "<<rt<<" nds "<<nds<<" lvl "<<lvl<<" is ";
    rt=crt(rt, nds/2, 0); pa[lvl][rt]=fa;
  //  cout<<rt<<el;
 
    bk[rt]=true; ct=0;
 
    // get ett
    cp[lvl][rt]=tp[lvl][rt]=rt;
    for(int &v:adj[rt]) if(!bk[v]) etr(v, rt, lvl, rt, v);
 
    // construct tree
    st[rt].init(nds);
    dfs(rt, 0);
 
    for(int &v:adj[rt]) if(!bk[v]) {
        rec(v, sz[v], rt, lvl+1);
    }
}
 
void qry(int u, int v, ll val) {

    // find smallest component
    int cur=0;
    while(cur<LOG-1&&cp[cur+1][u]!=0) ++cur;
  //  cout<<"iterating on "<<u<<" starting with "<<cur<<" delta "<<val<<el;
 
    // jump
    int crt=u;
    while(~cur) {
        // update edge
     //   cout<<"crt "<<crt<<el;
 
        if(cp[cur][u]==crt&&cp[cur][v]==crt) {
            int w=(in[cur][u]>in[cur][v]?u:v);
            ast.erase(ast.find(av[crt]));
 
           // cout<<"level "<<cur<<" w "<<w<<" cent "<<crt<<" updating "<<in[cur][w]<<" "<<ot[cur][w]<<el;
            st[crt].upd(1, 1, st[crt].wt, in[cur][w], ot[cur][w], val);
 
            // query max
            auto x=st[crt].rmq(1, 1, st[crt].wt, 1, st[crt].wt);
            x.s=mp[crt][x.s];
           // cout<<"x value "<<x.f<<" "<<x.s<<" "<<tp[cur][x.s]<<" "<<in[cur][tp[cur][x.s]]<<" "<<ot[cur][tp[cur][x.s]]<<el;
            auto y=max(st[crt].rmq(1, 1, st[crt].wt, 1, in[cur][tp[cur][x.s]]-1), st[crt].rmq(1, 1, st[crt].wt, ot[cur][tp[cur][x.s]]+1, st[crt].wt));
            //cout<<"y value "<<y.f<<" "<<x.s<<el;
 
            ll cnd=x.f+y.f; av[crt]=cnd;
            ast.insert(av[crt]);
        }
 
        crt=pa[cur--][crt];
    }
 
}
 
int32_t main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    cin>>N>>Q>>W;
    vector<pair<ll,pair<int,int>>> edg;
    memset(pa, -1, sizeof(pa));
    for(int i=1; i<N; ++i) {
        ll u, v, w; cin>>u>>v>>w;
        adj[u].pb(v); adj[v].pb(u);
        edg.pb({w,{u,v}});
    }
 
    // preprocess
    rec(1, N);
    for(int i=1; i<=N; ++i) ast.insert(0);
    for(auto &e:edg) qry(e.s.f, e.s.s, e.f);
 
    ll last=0, e, w;
    for(int q=1; q<=Q; ++q) {
        cin>>e>>w;
        e=(e+last)%(N-1);
        w=(w+last)%W;
      //  cout<<"actual qry "<<e<<" "<<w<<el;
        qry(edg[e].s.f, edg[e].s.s, w-edg[e].f);
        last=*ast.rbegin(); edg[e].f=w;
        cout<<last<<el;
    }
}
 
// MM

Compilation message

diameter.cpp: In member function 'void tree::build(int, int, int)':
diameter.cpp:38:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   38 |         int m=sl+sr>>1;
      |               ~~^~~
diameter.cpp: In member function 'void tree::upd(int, int, int, int, int, ll)':
diameter.cpp:54:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   54 |         int m=sl+sr>>1;
      |               ~~^~~
diameter.cpp: In member function 'std::pair<long long int, int> tree::rmq(int, int, int, int, int)':
diameter.cpp:64:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   64 |         int m=sl+sr>>1;
      |               ~~^~~

Subtask #1
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 18764 KB Output is correct
2 Correct 9 ms 18768 KB Output is correct
3 Correct 10 ms 18768 KB Output is correct
4 Correct 11 ms 18768 KB Output is correct
5 Correct 9 ms 18768 KB Output is correct
6 Correct 13 ms 18840 KB Output is correct
7 Correct 13 ms 18800 KB Output is correct
8 Correct 10 ms 18768 KB Output is correct
9 Correct 10 ms 18896 KB Output is correct
10 Correct 9 ms 18896 KB Output is correct
11 Correct 10 ms 18768 KB Output is correct
12 Correct 10 ms 18840 KB Output is correct
13 Correct 9 ms 18912 KB Output is correct
14 Correct 11 ms 18836 KB Output is correct
15 Correct 10 ms 18896 KB Output is correct
16 Correct 10 ms 18836 KB Output is correct
17 Correct 14 ms 18896 KB Output is correct
18 Correct 11 ms 18968 KB Output is correct

Subtask #2
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 18764 KB Output is correct
2 Correct 9 ms 18768 KB Output is correct
3 Correct 10 ms 18768 KB Output is correct
4 Correct 11 ms 18768 KB Output is correct
5 Correct 9 ms 18768 KB Output is correct
6 Correct 13 ms 18840 KB Output is correct
7 Correct 13 ms 18800 KB Output is correct
8 Correct 10 ms 18768 KB Output is correct
9 Correct 10 ms 18896 KB Output is correct
10 Correct 9 ms 18896 KB Output is correct
11 Correct 10 ms 18768 KB Output is correct
12 Correct 10 ms 18840 KB Output is correct
13 Correct 9 ms 18912 KB Output is correct
14 Correct 11 ms 18836 KB Output is correct
15 Correct 10 ms 18896 KB Output is correct
16 Correct 10 ms 18836 KB Output is correct
17 Correct 14 ms 18896 KB Output is correct
18 Correct 11 ms 18968 KB Output is correct
19 Correct 34 ms 20160 KB Output is correct
20 Correct 35 ms 20264 KB Output is correct
21 Correct 42 ms 20432 KB Output is correct
22 Correct 40 ms 20684 KB Output is correct
23 Correct 73 ms 25608 KB Output is correct
24 Correct 98 ms 27268 KB Output is correct
25 Correct 122 ms 28104 KB Output is correct
26 Correct 157 ms 29512 KB Output is correct

Subtask #3
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 18764 KB Output is correct
2 Correct 13 ms 18668 KB Output is correct
3 Correct 14 ms 18764 KB Output is correct
4 Correct 22 ms 18848 KB Output is correct
5 Correct 60 ms 19140 KB Output is correct
6 Correct 10 ms 18636 KB Output is correct
7 Correct 14 ms 18984 KB Output is correct
8 Correct 13 ms 19004 KB Output is correct
9 Correct 15 ms 18892 KB Output is correct
10 Correct 28 ms 19076 KB Output is correct
11 Correct 98 ms 19428 KB Output is correct
12 Correct 20 ms 21068 KB Output is correct
13 Correct 19 ms 21048 KB Output is correct
14 Correct 19 ms 21068 KB Output is correct
15 Correct 39 ms 21188 KB Output is correct
16 Correct 128 ms 21572 KB Output is correct
17 Correct 202 ms 65888 KB Output is correct
18 Correct 207 ms 65784 KB Output is correct
19 Correct 201 ms 65860 KB Output is correct
20 Correct 241 ms 65988 KB Output is correct
21 Correct 378 ms 66424 KB Output is correct

Subtask #4
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 20260 KB Output is correct
2 Correct 42 ms 20380 KB Output is correct
3 Correct 153 ms 20944 KB Output is correct
4 Correct 305 ms 21620 KB Output is correct
5 Correct 99 ms 36080 KB Output is correct
6 Correct 141 ms 36216 KB Output is correct
7 Correct 354 ms 36840 KB Output is correct
8 Correct 589 ms 37572 KB Output is correct
9 Correct 564 ms 119968 KB Output is correct
10 Correct 675 ms 120152 KB Output is correct
11 Correct 1037 ms 120872 KB Output is correct
12 Correct 1401 ms 121628 KB Output is correct
13 Correct 1292 ms 224752 KB Output is correct
14 Correct 1359 ms 225000 KB Output is correct
15 Correct 1817 ms 225484 KB Output is correct
16 Correct 2335 ms 226304 KB Output is correct
17 Correct 3312 ms 226168 KB Output is correct

Subtask #5
0 / 24.0

# Verdict Execution time Memory Grader output
1 Execution timed out 5050 ms 21952 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #6
0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 18764 KB Output is correct
2 Correct 9 ms 18768 KB Output is correct
3 Correct 10 ms 18768 KB Output is correct
4 Correct 11 ms 18768 KB Output is correct
5 Correct 9 ms 18768 KB Output is correct
6 Correct 13 ms 18840 KB Output is correct
7 Correct 13 ms 18800 KB Output is correct
8 Correct 10 ms 18768 KB Output is correct
9 Correct 10 ms 18896 KB Output is correct
10 Correct 9 ms 18896 KB Output is correct
11 Correct 10 ms 18768 KB Output is correct
12 Correct 10 ms 18840 KB Output is correct
13 Correct 9 ms 18912 KB Output is correct
14 Correct 11 ms 18836 KB Output is correct
15 Correct 10 ms 18896 KB Output is correct
16 Correct 10 ms 18836 KB Output is correct
17 Correct 14 ms 18896 KB Output is correct
18 Correct 11 ms 18968 KB Output is correct
19 Correct 34 ms 20160 KB Output is correct
20 Correct 35 ms 20264 KB Output is correct
21 Correct 42 ms 20432 KB Output is correct
22 Correct 40 ms 20684 KB Output is correct
23 Correct 73 ms 25608 KB Output is correct
24 Correct 98 ms 27268 KB Output is correct
25 Correct 122 ms 28104 KB Output is correct
26 Correct 157 ms 29512 KB Output is correct
27 Correct 11 ms 18764 KB Output is correct
28 Correct 13 ms 18668 KB Output is correct
29 Correct 14 ms 18764 KB Output is correct
30 Correct 22 ms 18848 KB Output is correct
31 Correct 60 ms 19140 KB Output is correct
32 Correct 10 ms 18636 KB Output is correct
33 Correct 14 ms 18984 KB Output is correct
34 Correct 13 ms 19004 KB Output is correct
35 Correct 15 ms 18892 KB Output is correct
36 Correct 28 ms 19076 KB Output is correct
37 Correct 98 ms 19428 KB Output is correct
38 Correct 20 ms 21068 KB Output is correct
39 Correct 19 ms 21048 KB Output is correct
40 Correct 19 ms 21068 KB Output is correct
41 Correct 39 ms 21188 KB Output is correct
42 Correct 128 ms 21572 KB Output is correct
43 Correct 202 ms 65888 KB Output is correct
44 Correct 207 ms 65784 KB Output is correct
45 Correct 201 ms 65860 KB Output is correct
46 Correct 241 ms 65988 KB Output is correct
47 Correct 378 ms 66424 KB Output is correct
48 Correct 17 ms 20260 KB Output is correct
49 Correct 42 ms 20380 KB Output is correct
50 Correct 153 ms 20944 KB Output is correct
51 Correct 305 ms 21620 KB Output is correct
52 Correct 99 ms 36080 KB Output is correct
53 Correct 141 ms 36216 KB Output is correct
54 Correct 354 ms 36840 KB Output is correct
55 Correct 589 ms 37572 KB Output is correct
56 Correct 564 ms 119968 KB Output is correct
57 Correct 675 ms 120152 KB Output is correct
58 Correct 1037 ms 120872 KB Output is correct
59 Correct 1401 ms 121628 KB Output is correct
60 Correct 1292 ms 224752 KB Output is correct
61 Correct 1359 ms 225000 KB Output is correct
62 Correct 1817 ms 225484 KB Output is correct
63 Correct 2335 ms 226304 KB Output is correct
64 Correct 3312 ms 226168 KB Output is correct
65 Execution timed out 5050 ms 21952 KB Time limit exceeded
66 Halted 0 ms 0 KB -