【dij变形】牛客练习赛93 C

最新推荐文章于 2024-09-17 16:22:09 发布

lamentropetion

最新推荐文章于 2024-09-17 16:22:09 发布

阅读量434

点赞数

分类专栏：图论文章标签： c语言开发语言

本文链接：https://blog.csdn.net/weixin_62528401/article/details/131676495

版权

图论专栏收录该内容

87 篇文章 0 订阅

订阅专栏

文章讲述了如何解决一个涉及C-点权的问题，使用了Dijkstra算法进行最短路径求解。代码中定义了图结构，使用了优先队列和multiset数据结构来优化松弛操作，确保在满足特定条件下开始松弛。算法首先初始化图，然后通过Dijkstra算法遍历节点，更新最短路径，并利用multiset存储每个节点的权值信息。

摘要由CSDN通过智能技术生成

C-点权_牛客练习赛93 (nowcoder.com)

题意：

思路：

重要的是在松弛的时候要满足什么条件才开始松弛

这里是用两个点来松弛一个点

Code：

#include <bits/stdc++.h>
 
//#define int long long
 
using namespace std;
 
const int mxn=1e5+10;
const int mxv=1e5+10;
const int mxe=1e5+10;
const int mod=1e9+7;
const int Inf=0x3f3f3f3f;

struct ty{
    int to,next,w;
}edge[mxe<<2];

struct ty2{
    int x,dis;
    bool operator<(const ty2&oth)const{
        return oth.dis<dis;
    }
};

multiset<int> S[mxn];

priority_queue<ty2> Q;

int N,u,v,w;
int tot=0;
int head[mxn],in[mxn];
int vis[mxn],dis[mxn];

void add(int u,int v,int w){
    edge[tot].w=w;
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void G_init(){
    tot=0;
    for(int i=0;i<=N;i++) head[i]=-1;
}
int get(int u){
    return (*S[u].begin())+(*(++S[u].begin()));
}
void dij(){
    memset(dis,0x3f,sizeof(dis));
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=N;i++){
        if(in[i]<=1){
            Q.push({i,0});
            dis[i]=0;
        }
        S[i].insert(Inf);
        S[i].insert(Inf);
    }
    while(!Q.empty()){
        auto u=Q.top();
        Q.pop();
        if(vis[u.x]) continue;
        vis[u.x]=1;
        for(int i=head[u.x];~i;i=edge[i].next){
            S[edge[i].to].insert(dis[u.x]+edge[i].w);
            if(dis[edge[i].to]>get(edge[i].to)){
                dis[edge[i].to]=get(edge[i].to);
                if(!vis[edge[i].to]) Q.push({edge[i].to,dis[edge[i].to]});
            }
        }
    }
}
void solve(){
    cin>>N;
    G_init();
    for(int i=1;i<=N-1;i++){
        cin>>u>>v>>w;
        add(u,v,w);
        add(v,u,w);
        in[u]++;
        in[v]++;
    }
    dij();
    for(int i=1;i<=N;i++){
        if(dis[i]==Inf) cout<<-1<<" \n"[i==N];
        else cout<<dis[i]<<" \n"[i==N];
    }
}
signed main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int __=1;//cin>>__;
    while(__--)solve();return 0;
}