题意:给一个数组arr,然后对于数组中1到n的位置pos,都可以跳向pos+arr[pos] 和pos-arr[pos],现在问题是:每一个位置跳向和它奇偶性相反的位置,最少需要几步。无法跳到则输出-1。
简单分析看题解分析:这个可以看成是对奇性点和偶性点跑多源最短路。那么,可以用超级源加上反向建边跑最短路 或者 是反向建边直接多源最短路跑bfs。
1.超级源,奇性超级源连接所有的奇性点,跑dij,那么对于从超级源到达偶性点的距离,就可以看成是偶性点到奇性点的最短路。同样,再建立偶性超级源。
2.多源最短路跑bfs。反向建边,直接将所有的奇性点入队,然后bfs,依次遍历到的偶性点 的 步数,即为结果。再同样跑一遍偶性点。
最后:代码
#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+100;
const int inf=0x3f3f3f3f;
int arr[maxn];
int ans[maxn];
vector<int>edge[maxn];
int n;
void solve(int flag)
{
int temp[maxn];
bool vis[maxn];
memset(temp,0,sizeof temp);
memset(vis,false,sizeof vis);
queue<int>q;
for(int i=1;i<=n;++i)
{
if((arr[i]&1)==flag)
{
q.push(i);
//cout<<"??"<<i<<endl;
vis[i]=true;
}
}
while(!q.empty())
{
int f=q.front();
q.pop();
//cout<<"pls"<<f<<endl;
for(int i=0;i<edge[f].size();++i)
{
if(!vis[edge[f][i]])
{
q.push(edge[f][i]);
vis[edge[f][i]]=true;
temp[edge[f][i]]=temp[f]+1;
}
}
}
for(int i=1;i<=n;++i)
{
if(vis[i]&&!((arr[i]&1)==flag))
ans[i]=temp[i];
}
}
int main()
{
cin>>n;
for(int i=1; i<=n; ++i)
cin>>arr[i];
for(int i=1;i<=n;++i)
{
if(i-arr[i]>=1)
{
edge[i-arr[i]].push_back(i);
}
if(i+arr[i]<=n)
{
edge[i+arr[i]].push_back(i);
}
}
solve(1);
solve(0);
for(int i=1;i<=n;++i)
{
if(ans[i])
cout<<ans[i]<<" ";
else
cout<<-1<<" ";
}
return 0;
}
乍一看,以为是dfs乱写一通就行了。然后wa10,就一直wa10,最后,在经历了无数的独立思索,在看了题解后,然后开始反思,终于才知道为什么自己写的麻瓜dfs 会wa,我有一个很简单的例子可以揭示错误,但是我要去玩了,就懒得写了。