QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#319429#2065. Cyclic DistanceTransparentTL 4ms38112kbC++204.3kb2024-02-02 16:10:032024-02-02 16:10:03

Judging History

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

  • [2024-02-02 16:10:03]
  • 评测
  • 测评结果:TL
  • 用时:4ms
  • 内存:38112kb
  • [2024-02-02 16:10:03]
  • 提交

answer

#include <bits/stdc++.h>

using namespace std;
using ll=long long;

constexpr int MAXN=4e5+10;

struct Node {
    int son[2],fa,siz;
    ll val,sum,tag;
    Node() {memset(son,0,sizeof(son)); fa=val=sum=siz=tag=0;}
};

struct Splay {
    Node t[MAXN<<1]; int stk[MAXN<<1],tot,top,rt[MAXN];
    Splay() {tot=top=0; t[0]=Node(); memset(rt,0,sizeof(rt));}
    int create(ll val=0) {
        int p=top?stk[top--]:++tot;
        t[p]=Node(); t[p].siz=1; t[p].val=t[p].sum=val; return p;
    }
    void drop(int x) {stk[++top]=x;}
    void clear(int n) {
        tot=0;
        memset(rt,0,sizeof(int)*(n+1));
    }
    bool isroot(int x) {return !t[x].fa;}
    bool getpos(int x) {return t[t[x].fa].son[1]==x;}
    void pushup(int x) {
        int ls=t[x].son[0],rs=t[x].son[1]; t[x].sum=t[x].val; t[x].siz=1;
        if(ls) t[x].sum+=t[ls].sum,t[x].siz+=t[ls].siz;
        if(rs) t[x].sum+=t[rs].sum,t[x].siz+=t[rs].siz;
    }
    void add(int x,ll v) {
        t[x].sum+=1ll*t[x].siz*v,t[x].val+=v,t[x].tag+=v;
    }
    void pushdown(int x) {
        int ls=t[x].son[0],rs=t[x].son[1];
        if(t[x].tag) {
            if(ls) add(ls,t[x].tag);
            if(rs) add(rs,t[x].tag);
            t[x].tag=0;
        }
    }
    void rotate(int x) {
        int y=t[x].fa,z=t[y].fa,p=getpos(x);
        if(!isroot(y)) t[z].son[getpos(y)]=x;
        t[x].fa=z; t[y].son[p]=t[x].son[p^1]; t[t[x].son[p^1]].fa=y;
        t[y].fa=x; t[x].son[p^1]=y; pushup(y); pushup(x);
    }
    void splay(int id,int x,int to=0) {
        for(int fa;fa=t[x].fa,!isroot(x);rotate(x)) if(!isroot(fa)) rotate(getpos(x)==getpos(fa)?fa:x);
        if(!to) rt[id]=x;
    }
    void insert(int id,ll v) {
        if(!rt[id]) {rt[id]=create(v); return;}
        int p=0,x=rt[id];
        while(x) pushdown(x),p=x,x=t[x].son[v<t[x].val];
        x=create(v); t[x].fa=p; t[p].son[v<t[p].val]=x; pushup(p); splay(id,x);
    }
    int findByOrder(int id,int ord) {
        assert(ord<=t[rt[id]].siz);  pushdown(rt[id]);
        for(int x=rt[id],ls,rs;ls=t[x].son[0],rs=t[x].son[1],x;pushdown(x)) {
            if(ord-t[ls].siz>0) {
                if(!(ord-=t[ls].siz+1)) {splay(id,x); return x;}
                x=rs;
            } else x=ls;
        }
        abort(); return -1;
    }
    void addl(int id,int r,int v) {
        if(r<1) return;
        int root=rt[id];
        if(t[root].siz<=r) {add(root,v); return;}
        int x=findByOrder(id,r+1); splay(id,x); pushdown(x);
        if(t[x].son[0]) add(t[x].son[0],v),pushdown(t[x].son[0]),splay(id,t[x].son[0]);
    }
    void addr(int id,int l,int v) {
        int root=rt[id];
        if(l>t[root].siz) return;
        if(l<=1) {add(root,v); return;}
        int x=findByOrder(id,l-1); splay(id,x); pushdown(x);
        if(t[x].son[1]) add(t[x].son[1],v),pushdown(t[x].son[1]),splay(id,t[x].son[1]);
    }
    ll queryl(int id,int r) {
        if(r<1) return 0;
        int root=rt[id];
        if(t[root].siz<=r) {return t[root].sum;}
        int x=findByOrder(id,r+1); splay(id,x); pushdown(x);
        return t[t[x].son[0]].sum;
    }
    void mergex(int id,int x) {
        pushdown(x);
        if(t[x].son[0]) mergex(id,t[x].son[0]);
        insert(id,t[x].val);
        if(t[x].son[1]) mergex(id,t[x].son[1]);
        drop(x);
    }
    void merge(int i1,int i2) {
        if(!rt[i1]) {rt[i1]=rt[i2]; rt[i2]=0; return;}
        if(!rt[i2]) return;
        if(t[rt[i1]].siz>t[rt[i2]].siz) mergex(i1,rt[i2]),rt[i2]=0;
        else mergex(i2,rt[i1]),rt[i1]=rt[i2],rt[i2]=0;
    }
    void print(int x) {
        pushdown(x);
        if(t[x].son[0]) print(t[x].son[0]);
        cout<<t[x].val<<" ";
        if(t[x].son[1]) print(t[x].son[1]);
    }
}t;

int n,k;
vector<pair<int,int>> g[MAXN];

void dfs(int u,int fa,int wf) {
    for(auto [v,w]:g[u]) if(v!=fa) dfs(v,u,w),t.merge(u,v);
    t.insert(u,0); int mid=k/2;
    if(k&1) t.addl(u,mid,wf),t.addr(u,mid+2,-wf);
    else t.addl(u,mid,wf),t.addr(u,mid+1,-wf);
}

void solve() {
    cin>>n>>k;
    for(int i=1;i<n;++i) {
        int u,v,w; cin>>u>>v>>w; g[u].emplace_back(v,w); g[v].emplace_back(u,w);
    }
    dfs(1,0,0);
    cout<<t.queryl(1,k)*2<<"\n";
    t.clear(n);
}

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

    int T=1; cin>>T;
    while(T--) solve();

    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

input:

1
5 3
1 2 4
1 3 1
1 4 8
4 5 9

output:

44

result:

ok single line: '44'

Test #2:

score: -100
Time Limit Exceeded

input:

57206
3 2
1 2 574927
2 3 406566
2 2
1 2 308806
4 3
1 2 312588
2 3 500141
2 4 53602
3 3
1 2 797183
2 3 944061
5 3
1 2 624010
2 3 488613
3 4 734514
4 5 497493
5 4
1 2 540278
2 3 488655
2 4 228989
2 5 653896
2 2
1 2 569117
4 2
1 2 821764
2 3 499471
1 4 549060
2 2
1 2 991159
2 2
1 2 482941
5 4
1 2 30462...

output:


result: