0pt TLE求助
查看原帖
0pt TLE求助
926212
std_qwq楼主2023/5/8 13:29

rt,所有数据要么AC要么TLE

#include<bits/stdc++.h>
#define x first
#define y second
#define int long long
using namespace std;
const int N=200010,M=19;
int f[N][M];
int h[N],e[2*N],ne[2*N],idx;
int dep[N];
int cnt[N],b[N];
int n;
pair<int,int> pr[N];
void add(int a,int b,int c1,int c2){
    e[idx]=b,ne[idx]=h[a],pr[idx]={c1,c2},h[a]=idx++;
}
void bfs(){
    queue<int> q;
    memset(dep,0x3f,sizeof dep);
    dep[1]=1,dep[0]=0;q.push(1);
    while(q.size()){
        int x=q.front();q.pop();
        for(int i=h[x];i!=-1;i=ne[i]){
            int y=e[i];
            if(dep[y]>dep[x]+1){
                dep[y]=dep[x]+1;
                f[y][0]=x;
                for(int k=1;k<19;k++)f[y][k]=f[f[y][k-1]][k-1];
                q.push(y);
            }
        }
    }
}
int lca(int a,int b){
    if(dep[a]<dep[b])swap(a,b);
    for(int i=18;i>=0;i--)if(dep[f[a][i]]>=dep[b])a=f[a][i];
    if(a==b)return a;
    for(int i=18;i>=0;i--)
        if(f[a][i]!=f[b][i]){
            a=f[a][i];
            b=f[b][i];
        }
    return f[a][0];
}
int dfs(int u,int fa){
    int c=b[u];
    for(int i=h[u];i!=-1;i=ne[i]){
        int v=e[i];
        if(v==fa)continue;
        c+=dfs(v,u);
        cnt[i]=dfs(v,u);
    }
    return c;
}
signed main(){
    cin>>n;
    memset(h,-1,sizeof h);
    for(int i=1;i<n;i++){
        int a,b,c1,c2;
        cin>>a>>b>>c1>>c2;
        add(a,b,c1,c2),add(b,a,c1,c2);
    }
    bfs();
    for(int i=2;i<=n;i++){
        int p=lca(i-1,i);
        b[i]++,b[i-1]++,b[p]-=2;
    }
    dfs(1,-1);
    int res=0;
    for(int i=0;i<idx;i++){
        int w;
        w=min(pr[i].x*cnt[i],pr[i].y);
        res+=w;
    }
    cout<<res;
}
2023/5/8 13:29
加载中...