1272E
题意:给一个数组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]<<" ";elsecout<<-1<<" ";}return 0;
}
乍一看,以为是dfs乱写一通就行了。然后wa10,就一直wa10,最后,在经历了无数的独立思索,在看了题解后,然后开始反思,终于才知道为什么自己写的麻瓜dfs 会wa,我有一个很简单的例子可以揭示错误,但是我要去玩了,就懒得写了。