欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Codeforces 954D. Fight Against Traffic

程序员文章站 2022-03-08 15:31:22
原题链接#pragma GCC optimize(2)#includeusing namespace std;typedef long long ll;typedef unsigned long long ull;typedef pairPLL;typedef pairPII;typedef pairPDD;#define I_int llinlin...

原题链接
题意:
给出n点m边的无向图,现在要求在没有边的两点之前连边,并且s到t的最短路长度不变,问满足条件的边数。
思路:
从起点和终点分别跑一遍最短路,然后枚举一下未出现的边统计即可。

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll>PLL;
typedef pair<int,int>PII;
typedef pair<double,double>PDD;
#define I_int ll
inline ll read()
{
    ll x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
#define read read()
#define closeSync ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define multiCase int T;cin>>T;for(int t=1;t<=T;t++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i<(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define perr(i,a,b) for(int i=(a);i>(b);i--)
ll ksm(ll a,ll b,ll p)
{
    ll res=1;
    while(b)
    {
        if(b&1)res=res*a%p;
        a=a*a%p;
        b>>=1;
    }
    return res;
}
#define PI acos(-1)
#define x first
#define y second
const int maxn=1e6+7,inf=0x3f3f3f3f;
int n,m,s,t;
struct node{
    int e,ne;
}edge[maxn];
int h[maxn],idx;
void add(int u,int v){
    edge[idx]={v,h[u]},h[u]=idx++;
}
int dis[maxn],st[maxn],dis1[maxn];
void dijkstra(int s){
    memset(dis,0x3f,sizeof dis);
    memset(st,0,sizeof st);
    dis[s]=0;
    ///建立一个维护最小值的优先队列
    priority_queue<PII,vector<PII>,greater<PII>>heap;
    heap.push({0,s});///起始点放入队列
    while(heap.size()){
        auto t=heap.top();///最小值
        heap.pop();
        int ver=t.second,d=t.first;
        if(st[ver]) continue;///该点更新
        st[ver]=true;
        for(int i=h[ver];i!=-1;i=edge[i].ne){
            int j=edge[i].e;
            if(dis[j]>d+1){
                dis[j]=d+1;
                heap.push({dis[j],j});
            }
        }
    }
}

void dijkstra1(int s){
    memset(dis1,0x3f,sizeof dis1);
    memset(st,0,sizeof st);
    dis1[s]=0;
    ///建立一个维护最小值的优先队列
    priority_queue<PII,vector<PII>,greater<PII>>heap;
    heap.push({0,s});///起始点放入队列
    while(heap.size()){
        auto t=heap.top();///最小值
        heap.pop();
        int ver=t.second,d=t.first;
        if(st[ver]) continue;///该点更新
        st[ver]=true;
        for(int i=h[ver];i!=-1;i=edge[i].ne){
            int j=edge[i].e;
            if(dis1[j]>d+1){
                dis1[j]=d+1;
                heap.push({dis1[j],j});
            }
        }
    }
}
int g[1100][1100];
int main()
{
    memset(h,-1,sizeof h);idx=0;
    n=read,m=read,s=read,t=read;
    rep(i,1,m){
        int u=read,v=read;
        add(u,v);add(v,u);
        g[u][v]=g[v][u]=1;
    }
    dijkstra(s);
    dijkstra1(t);
    int res=0;
    for(int i=1;i<n;i++)
        for(int j=i+1;j<=n;j++)
            if(!g[i][j]){
                int d1=dis[i]+dis1[j]+1,d2=dis[j]+dis1[i]+1;
                if(d1>=dis[t]&&d2>=dis[t]) res++;
            }
    printf("%d\n",res);
    return 0;
}

本文地址:https://blog.csdn.net/weixin_45675097/article/details/114251161

相关标签: 笔记