CodeForces - 1272E Nearest Opposite Parity(多源起点的最短路)

题目链接：点击查看

题目大意：给出 n 个点，每个点都有一个权值 a[ i ]，且每个点可以到达 ( i - a[ i ] ) 和 ( i + a[ i ] ) 两个位置，花费为 1，问每个点到达与其本身奇偶不同的点的最小花费是多少

题目分析：看到网上那么多假算法的题解真的人傻了，写了个 spfa 恩说是 bfs ，不过这题好像真的没法卡掉 spfa，但也不能指鹿为马吧

因为每个偶数权值的点的终点是奇数权值，同理每个奇数权值的点的终点是偶数权值，如此一来可以反向建边，按照奇偶分别跑一次多源起点的最短路就是答案了

代码：
 

// #pragma GCC optimize(2)
// #pragma GCC optimize("Ofast","inline","-ffast-math")
// #pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
#include<unordered_map>
using namespace std;
     
typedef long long LL;
     
typedef unsigned long long ull;
     
const int inf=0x3f3f3f3f;

const int N=2e5+100;//顶点数 
 
const int M=1e6+100;//边数 
 
struct Edge
{
	int to,w,next;
}edge[M];
 
int head[N],d[N],a[N],ans[N],cnt,n;//链式前向星 
 
bool vis[N];
 
void addedge(int u,int v,int w)
{
	edge[cnt].to=v;
	edge[cnt].w=w;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
 
struct Node
{
	int to,w;
	Node(int TO,int W)
	{
		to=TO;
		w=W;
	}
	bool operator<(const Node& a)const
	{
		return w>a.w;
	}
};
 
void Dijkstra(int flag)
{
	priority_queue<Node>q;
	memset(vis,false,sizeof(vis));
	memset(d,inf,sizeof(d));
	for(int i=1;i<=n;i++)
		if(a[i]%2==flag)
		{
			q.push(Node(i,0));
			d[i]=0;
		}
	while(q.size())
	{
		Node cur=q.top();
		int u=cur.to;
		q.pop();
		if(vis[u])
			continue;
		vis[u]=true;
		for(int i=head[u];i!=-1;i=edge[i].next)//扫描出所有边 
		{
			int v=edge[i].to;
			int w=edge[i].w;
			if(d[v]>d[u]+w)//更新 
			{
				d[v]=d[u]+w;
				q.push(Node(v,d[v]));
			}
		}
	}
	for(int i=1;i<=n;i++)
		if(a[i]%2!=flag&&d[i]!=inf)
			ans[i]=d[i];
}
 
void init()
{
	memset(ans,-1,sizeof(ans));
	memset(head,-1,sizeof(head));
	cnt=0; 
}
 
int main()
{
#ifndef ONLINE_JUDGE
//  freopen("data.in.txt","r",stdin);
//  freopen("data.out.txt","w",stdout);
#endif
//  ios::sync_with_stdio(false);
	init();
 	scanf("%d",&n);
 	for(int i=1;i<=n;i++)
 	{
 		scanf("%d",a+i);
 		if(i-a[i]>=1)
 			addedge(i-a[i],i,1);
 		if(i+a[i]<=n)
 			addedge(i+a[i],i,1);
 	}
 	Dijkstra(0);
 	Dijkstra(1);
 	for(int i=1;i<=n;i++)
 		printf("%d ",ans[i]);
 	puts("");
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
    return 0;
}

标签：Nearest,cnt,Parity,int,1272E,long,edge,权值,include
来源： https://blog.csdn.net/qq_45458915/article/details/111200940