Jzzhu and Cities 【CodeForces - 449B】【最短路条数】

题目链接

题目大意

现在有n个城市标号1–n，有m条双向公路路连接，每条路有一个长度，还有k条火车，每个火车相连1到xi，每个火车路线有一个长度。问你在不破坏1到任何城市的最短路的情况下，最多能撤掉多少个火车路线。

解题思路

我们在跑最短路的同时，用一个vis数组来标记这个最短路的条数有多少个。当从1到第k个城市的最短路有一个的时候，那么这个火车路线就不能被撤离，如果有多个的话，就可以被撤离。
这个题的数据量太大，所以在记录最短路条数的时候，爆long long了，所以我最后取余了一下。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;

int n,m,k,tot;
const int N=1e5+7;
int f[N];
struct qnode
{
    int v,c;
    qnode(int V,int C)
    {
        v=V;
        c=C;
    }
    qnode() {}
    bool operator <(const qnode &r)const
    {
        return c>r.c;
    }
};
struct node
{
    int v,nex;
    long long w;
};
node e[N*8];
long long dis[N],b[N],vis[N];
int book[N],a[N];
const long long inf=9223372036854775806;

void add(int u,int v,long long w)
{
    e[tot].v=v;
    e[tot].w=w;
    e[tot].nex=f[u];
    f[u]=tot++;
}
void dij()
{
    memset(book,0,sizeof(book));
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)
        dis[i]=inf;
    priority_queue<qnode>que;
    while(!que.empty())
     que.pop();

     dis[1]=0;
     vis[1]=1;
     que.push(qnode(1,0));
     qnode tmp;
     while(!que.empty())
     {
         tmp=que.top();
         que.pop();
         int u=tmp.v;
         if(book[u])
            continue;
         book[u]=1;
         for(int i=f[u];i!=-1;i=e[i].nex)
         {
             int v=e[i].v;
             if(!book[v]&&dis[v]>dis[u]+e[i].w)
             {
                 vis[v]=vis[u];
                 dis[v]=dis[u]+e[i].w;
                 que.push(qnode(v,dis[v]));
             }
             else if(!book[v]&&dis[v]==dis[u]+e[i].w)
                vis[v]=(vis[v]+vis[u])%(long long)10000000007;
         }
     }
}
int main()
{
    memset(f,-1,sizeof(f));
    scanf("%d %d %d",&n,&m,&k);
    int x,y;
    long long z;
    for(int i=1;i<=m;i++)
    {
        scanf("%d %d %lld",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    for(int i=1;i<=k;i++)
    {
        scanf("%d %d",&a[i],&b[i]);
        add(1,a[i],b[i]);
        add(a[i],1,b[i]);
    }
    dij();
    int ans=0;
    for(int i=1;i<=k;i++)
    {
        if(dis[a[i]]<b[i])
            ans++;
        else if(dis[a[i]]==b[i]&&vis[a[i]]>1)
        {
            vis[a[i]]--;
            ans++;
        }
    }
    printf("%d\n",ans);
    return 0;
}