萌新刚学OI,求助16pts MLE
查看原帖
萌新刚学OI,求助16pts MLE
475112
MarSer020楼主2023/11/13 10:34

rt,我就写个双倍经验怎么你了((

除了 #1 #2 全部 MLE

代码:

#include<bits/stdc++.h>
using namespace std;
int n,m,q,fa[100005][18],maxv[100005][18],fat[100005],cnt,dep[100005],x,y;
pair<int,pair<int,int>>a[100005];
vector<pair<int,int>>e[100005];
int find(int x){
    return x==fat[x]?x:fat[x]=find(fat[x]);
}
void dfs(int cur,int fat){
    fa[cur][0]=fat,dep[cur]=dep[fat]+1;
    for(auto x:e[cur])
        if(x.first!=fat)
            maxv[x.first][0]=x.second,dfs(x.first,cur);
}
int fmxv(int x,int y){
    int ret=0;
    if(dep[x]<dep[y])
        swap(x,y);
    for(int i=17;i>=0;i--)
        if(dep[fa[x][i]]>=dep[y])
            ret=max(ret,maxv[x][i]),x=fa[x][i];
    if(x==y)
        return ret;
    for(int i=17;i>=0;i--)
        if(fa[x][i]!=fa[y][i])
            ret=max(ret,max(maxv[x][i],maxv[y][i])),x=fa[x][i],y=fa[y][i];
    return max(ret,max(maxv[x][0],maxv[y][0]));
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin>>n>>m;
    for(int i=1;i<=m;i++)
        cin>>a[i].second.first>>a[i].second.second>>a[i].first;
    sort(a+1,a+m+1);
    for(int i=1;i<=n;i++)
        fat[i]=i;
    for(int i=1,x,y;i<=m;i++){
        x=a[i].second.first,y=a[i].second.second;
        if(find(x)!=find(y))
            e[x].emplace_back(make_pair(y,a[i].first)),e[y].emplace_back(make_pair(x,a[i].first)),fat[y]=x;
    }
    for(int i=1;i<=n;i++)
        if(!fa[i][0])
            dfs(i,i);
    for(int j=1;j<=17;j++)
        for(int i=1;i<=n;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1],maxv[i][j]=max(maxv[i][j-1],maxv[fa[i][j-1]][j-1]);
    cin>>q;
    while(q--){
        cin>>x>>y;
        if(find(x)!=find(y))
            cout<<"impossible\n";
        else
            cout<<fmxv(x,y)<<'\n';
    }
    return 0;
}
2023/11/13 10:34
加载中...