oj.uz

Submission #711141

# Submission time Handle Problem Language Result Execution time Memory
711141 2023-03-16T08:56:19 Z TimDee Paths (RMI21_paths) C++17
31 / 100
 515 ms 321868 KB 
#include <bits/stdc++.h>
using namespace std;
 
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx2,popcnt")

using ll = long long;
#define int long long
#define double long double
#define forn(i,n) for(int i=0; i<(n); ++i)
#define pb push_back
#define pi pair<int,int>
#define f first
#define s second 
#define vii(a,n) vector<int> a(n); forn(i,n) cin>>a[i];
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const int inf = 1e18;
const int mod = 1e9+7;//998244353;

const int N = 1e5;
vector<pi> adj[N];
vector<pi> z;
vector<pi> mx(N,{-1,-1});
vector<pi> out(N);
int ans[N];

void dfs(int u, int p) {
    for (auto&e:adj[u]) {
        int v=e.f, w=e.s;
        if (v==p) continue;
        dfs(v,u);
        if (mx[v].f+w > mx[u].f) {
            if (mx[u].s!=-1) z.pb(mx[u]);
            mx[u]={mx[v].f+w,mx[v].s};
        } else {
            z.pb({mx[v].f+w,mx[v].s});
        }
    }
    if (adj[u].size()==1) mx[u]={0,u};
}
void dfs2(int u, int p, pi o) {
    out[u]=o;
    vector<pi> pr={{-1,-1}};
    for(auto&e:adj[u]) {
        int v=e.f, w=e.s;
        if (v==p) {
            pr.pb(pr.back());
            continue;
        }
        if (mx[v].f+w>pr.back().f) pr.pb({mx[v].f+w,mx[v].s});
        else pr.pb(pr.back());
    }
    pi sf={-1,-1};
    int sz=adj[u].size();
    for (int i=sz-1; i>=0; --i) {
        auto e=adj[u][i];
        int v=e.f, w=e.s;
        if (v==p) continue;
        auto x=(pr[i].f > sf.f)?pr[i]:sf;
        if (o.f > x.f) x=o;
        x.f+=w;
        dfs2(v,u,x);
        if (mx[v].f+w > sf.f) sf={mx[v].f+w,mx[v].s};
    }
}

const int sz=1ll<<47;

struct sgt {
    int l,r;
    int c,s;
    sgt() {
        l=r=-1;
        c=s=0;
    }
};

const int tsz=9.6e6;
sgt t[tsz];
int nxt=1;

void add(int v, int l, int r, int x) {
    t[v].c++, t[v].s+=x;
    if (r-l==1) return;
    int m=(l+r)>>1;
    int L=t[v].l, R=t[v].r;
    if (x<m) {
        if (L==-1) t[v].l=nxt++;
        add(t[v].l,l,m,x);
    } else {
        if (R==-1) t[v].r=nxt++;
        add(t[v].r,m,r,x);
    }
}
void add(int x) {
    add(0,0,sz,x);
}
void del(int v, int l, int r, int x) {
    t[v].c--, t[v].s-=x;
    if (r-l==1) return;
    int m=(l+r)>>1;
    int L=t[v].l, R=t[v].r;
    if (x<m) {
        assert (L!=-1);
        del(t[v].l,l,m,x);
    } else {
        assert (R!=-1);
        del(t[v].r,m,r,x);
    }
}
void del(int x) {
    del(0,0,sz,x);
}

int query(int v, int l, int r, int k) {
    int c=t[v].c, s=t[v].s;
    if (!k) return 0;
    if (c==k) return s;
    assert(k<=c);
    if (r-l==1) return k*l;
    int ret=0;
    int m=(l+r)>>1;
    int L=t[v].l, R=t[v].r;
    if (R!=-1) {
        if (t[R].c < k) {
            k-=t[R].c;
            ret+=t[R].s;
        } else return query(R,m,r,k);
    }
    assert(L!=-1);
    return ret+query(L,l,m,k);
}
int query(int k) {
    return query(0,0,sz,k);
}

int n,k;
int Z[N];
void dfs3(int u, int p) {
    ans[u]=query(k);
    for(auto&e:adj[u]) {
        int v=e.f, w=e.s;
        if (v==p) continue;
        del(Z[out[v].s]);
        Z[out[v].s]+=w;
        add(Z[out[v].s]);
        del(Z[mx[v].s]);
        Z[mx[v].s]-=w;
        add(Z[mx[v].s]);
        dfs3(v,u);
        del(Z[out[v].s]);
        Z[out[v].s]-=w;
        add(Z[out[v].s]);
        del(Z[mx[v].s]);
        Z[mx[v].s]+=w;
        add(Z[mx[v].s]);
    }
}

void solve() {

	cin>>n>>k;
    forn(i,n-1) {
        int u,v,w; cin>>u>>v>>w; --u, --v;
        adj[u].pb({v,w});
        adj[v].pb({u,w});
    }
    dfs(0,-1);
    z.pb(mx[0]);
    pi puiu1={0,0}, puiu2={-1,-1};
    pi x = (adj[0].size()==1)?puiu1:puiu2;
    dfs2(0,-1,x);

    if (z.size()<=k) {
        int s=0;
        for(auto&x:z) s+=x.f;
        forn(i,n) cout<<s<<'\n';
    }
    for(auto&x:z) {
        Z[x.s]=x.f;
        add(x.f);
    }
    dfs3(0,-1);
    forn(i,n) cout<<ans[i]<<'\n';

}
     
int32_t main() {
	ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int t=1;
    //cin>>t;
    while (t--) solve();
    return 0;
}

Compilation message

Main.cpp: In function 'void solve()':
Main.cpp:178:17: warning: comparison of integer expressions of different signedness: 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
  178 |     if (z.size()<=k) {
      |         ~~~~~~~~^~~

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 143 ms 306332 KB Output is correct
2 Correct 136 ms 306376 KB Output is correct

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 143 ms 306332 KB Output is correct
2 Correct 136 ms 306376 KB Output is correct
3 Correct 138 ms 306416 KB Output is correct
4 Correct 143 ms 306504 KB Output is correct
5 Correct 148 ms 306304 KB Output is correct
6 Correct 133 ms 306380 KB Output is correct
7 Correct 133 ms 306388 KB Output is correct

Subtask #3
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 143 ms 306332 KB Output is correct
2 Correct 136 ms 306376 KB Output is correct
3 Correct 138 ms 306416 KB Output is correct
4 Correct 143 ms 306504 KB Output is correct
5 Correct 148 ms 306304 KB Output is correct
6 Correct 133 ms 306380 KB Output is correct
7 Correct 133 ms 306388 KB Output is correct
8 Incorrect 137 ms 306552 KB Output isn't correct
9 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 143 ms 306332 KB Output is correct
2 Correct 136 ms 306376 KB Output is correct
3 Correct 138 ms 306416 KB Output is correct
4 Correct 143 ms 306504 KB Output is correct
5 Correct 148 ms 306304 KB Output is correct
6 Correct 133 ms 306380 KB Output is correct
7 Correct 133 ms 306388 KB Output is correct
8 Incorrect 137 ms 306552 KB Output isn't correct
9 Halted 0 ms 0 KB -

Subtask #5
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 501 ms 315060 KB Output is correct
2 Correct 505 ms 321868 KB Output is correct
3 Correct 479 ms 316532 KB Output is correct
4 Correct 499 ms 317428 KB Output is correct
5 Correct 500 ms 318988 KB Output is correct
6 Correct 515 ms 317012 KB Output is correct

Subtask #6
0 / 32.0

# Verdict Execution time Memory Grader output
1 Correct 143 ms 306332 KB Output is correct
2 Correct 136 ms 306376 KB Output is correct
3 Correct 138 ms 306416 KB Output is correct
4 Correct 143 ms 306504 KB Output is correct
5 Correct 148 ms 306304 KB Output is correct
6 Correct 133 ms 306380 KB Output is correct
7 Correct 133 ms 306388 KB Output is correct
8 Incorrect 137 ms 306552 KB Output isn't correct
9 Halted 0 ms 0 KB -