CF 449B Jzzhu and Cities (单源最短路SPFA)

最新推荐文章于 2021-01-25 09:52:34 发布

M_Y_Y_

最新推荐文章于 2021-01-25 09:52:34 发布

阅读量194

点赞数

分类专栏：图论文章标签： SPFA 单源最短路

本文链接：https://blog.csdn.net/M_Y_Y_/article/details/97622751

版权

图论专栏收录该内容

7 篇文章 0 订阅

订阅专栏

codeforces 449B

题意：n个城市，从1～n编号，m条公路，k条铁路，铁路都是连接城市1和城市s（s由输入得到）的，问最多拆除几条铁路，各个城市到城市1的距离仍最短。

思路：数据量有点大，我用的SPFA，算出包括所有铁路公路在内的城市1到其他城市的最短路，然后后期删除不需要的铁路，用一个数组记录每个点的最短路的条数，主要是为了记录到该点长度相同的道路有几条，即入度，方便删除铁路。

刚开始的思路是，计算出只包含公路的单源最短路，然后和铁路比较，这种思路是错的QAQ，比如1 -> 3 -> 4 -> 2 -> 5 是不加铁路的一条最短路，一条铁路为1 -> 2 ，然后1到5的最短距离可能是1 -> 2 -> 5，所以计算是应该把铁路也包含进去。

代码：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <sstream>
#include <cstdio>
#include <vector>
#include <string>
#include <cmath>
#include <stack>
#include <queue>
#include <map>
#include <set>
#define INF 0x3f3f3f3f
#define fori(a,b) for(LL i=a;i<=b;i++)
#define forj(a,b) for(LL j=a;j<=b;j++)
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long LL;
const double PI = acos(-1);
const LL M = 1e5+10;
//const LL MM = 1e5+10;
LL n,m,k;
LL dis[M];
LL vis[M];
LL ru[M];
struct node{
    LL en,w;
    LL flag;
}st,tl[M];
queue<LL>q;
vector<node>v[M];
void s()
{
    q.push(1);
    mem(vis,0);
    mem(ru,0);
    mem(dis,INF);
    dis[1]=0;
    vis[1]=1;
    while(!q.empty()){
        LL now=q.front();
        q.pop();
        vis[now]=0;                                      //出队列清0
        for(LL i=0;i<v[now].size();i++){
            if(dis[v[now][i].en]>dis[now]+v[now][i].w){
                ru[v[now][i].en]=1;                      //入度清为1
                dis[v[now][i].en]=dis[now]+v[now][i].w;  //更新路径
                if(!vis[v[now][i].en]){
                    q.push(v[now][i].en);                //该点入队列
                    vis[v[now][i].en]=1;
                }
            }
            else if(dis[v[now][i].en]==dis[now]+v[now][i].w) //相等加1
                    ru[v[now][i].en]++;

        }

    }
    return ;
}
int main()
{
    //freopen("//home//acm//桌面//in","r",stdin);
    map<LL ,map<LL,LL> >::iterator it;
    LL a,b,w;
    cin >> n >> m >> k;
    fori(1,m){
        cin >> a >> b >> w;
        st.en=b,st.w=w;
        v[a].push_back(st);
        st.en=a,st.w=w;
        v[b].push_back(st);
    }
    for(LL i=1;i<=k;i++){
        cin >> b >> w;
        tl[i].en=b,tl[i].w=w;
        v[1].push_back(tl[i]);
    }
    s();
    LL ans=0;
    for(LL i=1;i<=k;i++){
        if(dis[tl[i].en]<tl[i].w)
            ans++;
        //相等的铁路删去一条，入度减1
        if(dis[tl[i].en]==tl[i].w&&ru[tl[i].en]>1){  
            ru[tl[i].en]--;
            ans++;
        }
    }
    cout << ans << endl;
    return 0;
}