题目链接:点击查看
题目大意:给出 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;
}