cf 507E Breaking Good 最短路

在一个有n个城市m条边的国家,有一个犯罪团伙。这个犯罪团伙想要抢银行。犯罪团伙基地在城市1,银行在城市n。有个人Walter想要加入这个犯罪团伙,于是为了过的团伙高层的信任,Walter必须担任一项的任务,那就是在基地到银行之间选出一条最短路。这个国家有若干公路在维修,于是你需要将最短路径上的公路修好,为了阻止别的警察追击我方,还必须将所有的不在最短路上的公路炸毁。假如有多条最短路,那么需要选择影响值最小的一条最短路(影响值=需要维修的公路数+需要炸毁的公路数).

扩大边权的技巧,一开始影响值是m,走一条完好的路影响值减一,走一条坏路影响值加一,所以即求一条最短路,若有多条最短路那么走完好的路最多的。

#include <cstdio>
#include <cstring>
#include <queue>
#define ll long long
using namespace std;
#define maxn 101000
const ll inf = 0x3f3f3f3f3f3f3f;
int s[maxn],t[maxn],is[maxn];
int head[maxn],cnt;
int pre[maxn],preE[maxn];
int n,m;
ll dist[maxn];
int vis[maxn];
struct node
{
    int u,v,id,next;
    ll w;
}e[maxn*2];

void add(int u,int v,int id,ll w)
{
    e[cnt].u=u;
    e[cnt].v=v;
    e[cnt].id=id;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt++;
}

void spfa()
{
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++) dist[i]=inf;
    queue <int> Q;
    dist[1]=0;
    Q.push(1);
    memset(pre,-1,sizeof(pre));
    while(!Q.empty())
    {
        int u=Q.front(); Q.pop(); vis[u]=0;
        for(int i=head[u];i!=-1;i=e[i].next)
        {
            int v=e[i].v;
            if(dist[v] > dist[u] + e[i].w)
            {
                dist[v]=dist[u]+e[i].w;
                pre[v]=u;
                preE[v]=e[i].id;
                if(!vis[v])
                {
                    Q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    int x,y,z;
    cnt=0;
    memset(head,-1,sizeof(head));
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        s[i]=x;t[i]=y;is[i]=z;
        add(x,y,i,maxn+1-z);
        add(y,x,i,maxn+1-z);
    }
    spfa();
    memset(vis,0,sizeof(vis));
    for(int i=n;pre[i]!=-1;i=pre[i])
    {
        int id=preE[i];
        if(is[id]==0)   vis[id]=2;
        if(is[id]==1)   vis[id]=1;
    }
    for(int i=0;i<m;i++)
        if(vis[i]==0&&is[i]==1)
            vis[i]=2;
    int tot=0;
    for(int i=0;i<m;i++)
        if(vis[i]==2) tot++;
    printf("%d\n",tot);
    for(int i=0;i<m;i++)
        if(vis[i]==2)
            printf("%d %d %d\n",s[i],t[i],1-is[i]);
    return 0;
}


你可能感兴趣的:(cf 507E Breaking Good 最短路)