P3393 逃离僵尸岛

Portal.

建一个虚拟节点 0 0 0 号节点(一种很常用的思路),把这个点与所有被僵尸占领的点都连一条边权为 1 1 1 的边,跑一遍从 0 0 0 开始的 Dijkstra,对于一个点如果得到的 dis 数组值小于等于 S + 1 S+1 S+1,那么就是危险城市。

然后跑一遍 Dijkstra,如果到达点为危险城市,边权为 Q Q Q;如果到达点为起点或终点,边权为 0 0 0;如果到达点为僵尸占领的点时,边权为 + ∞ +\infty +∞。

代码如下:

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define int long long

const int maxn=600005;
int cnt,dis[maxn],a[maxn],b[maxn],head[maxn],nxt[maxn],to[maxn],w[maxn],N,M,K,S,c[maxn],P,Q;
bool danger[maxn],vis[maxn];

void add(int x,int y,int z)
{
    to[++cnt]=y;
    w[cnt]=z;
    nxt[cnt]=head[x];
    head[x]=cnt;
}

struct node
{
	int id,dis;
	friend bool operator < (node a,node b)
	{
		return a.dis>b.dis;
	}
};

priority_queue<node> q;

void dij(int s)
{
	memset(vis,0,sizeof(vis));
	memset(dis,0x3f3f3f,sizeof(dis));
	dis[s]=0;q.push(node{s,0});
	while(!q.empty())
	{
		int u=q.top().id;q.pop();
		if(vis[u]) continue;
		vis[u]=1;
		for(int i=head[u];i;i=nxt[i])
			if(dis[to[i]]>dis[u]+w[i]&&!vis[to[i]])
				dis[to[i]]=dis[u]+w[i],q.push(node{to[i],dis[to[i]]});
	}
}

signed main()
{
	cin>>N>>M>>K>>S;
	cin>>P>>Q;
	for(int i=1;i<=K;i++) cin>>c[i],add(0,c[i],1),add(c[i],0,1);
	for(int i=1;i<=M;i++) cin>>a[i]>>b[i],add(a[i],b[i],1),add(b[i],a[i],1);
	dij(0);
	for(int i=1;i<=N;i++) if(dis[i]<=S+1) danger[i]=1;
	cnt=0;
	for(int i=1;i<=M;i++) head[i]=0,nxt[i]=0,w[i]=0,vis[i]=0;
	for(int i=1;i<=K;i++) vis[c[i]]=1;
	int tmp;
	for(int i=1;i<=M;i++)
	{
		if(vis[a[i]]||vis[b[i]]) continue;
		if(danger[b[i]]) tmp=Q;
		else tmp=P;
		add(a[i],b[i],tmp);
		if(danger[a[i]]) tmp=Q;
		else tmp=P;
		add(b[i],a[i],tmp);
	}
	dij(1);
	if(!danger[N]) cout<<dis[N]-P;
	else cout<<dis[N]-Q;
	return 0;
}
相关推荐
希望20171 天前
图论基础知识
算法·图论
行走的bug...2 天前
用图论来解决问题
算法·图论
Athenaand2 天前
代码随想录算法训练营第50天 | 图论理论基础、深搜理论基础、98. 所有可达路径、广搜理论基础
算法·图论
_Coin_-3 天前
算法训练营DAY60 第十一章:图论part11
算法·图论
Athenaand3 天前
代码随想录算法训练营第62天 | Floyd 算法精讲、A * 算法精讲 (A star算法)、最短路算法总结篇、图论总结
算法·图论
ZLRRLZ4 天前
【数据结构】图
数据结构·算法·图论
一只鱼^_8 天前
牛客周赛 Round 108
数据结构·c++·算法·动态规划·图论·广度优先·推荐算法
_Coin_-8 天前
算法训练营DAY58 第十一章:图论part08
数据结构·算法·图论
闪电麦坤9511 天前
数据结构:图的表示 (Representation of Graphs)
数据结构·算法·图论
BlackPercy11 天前
【图论】Graphs.jl 最小生成树算法文档
算法·图论