E. Nearest Opposite Parity
题意:从一点至少要走几次才能走到奇偶性不同的点
建立0,n+1超级源点,分别表示奇数点,偶数点
这里的dijkstra --利用了大根堆(优先队列) —第一位为dis的相反数利用相反数把他变成小根堆
#include<bits/stdc++.h>
using namespace std;
const int N=200010;
const int inf=0x3f3f3f3f;
vector<pair<int,int> >v[N];
int vis[N],d[N],n,a[N],ans[N];
void dijkstra(int x)
{
for(int i=0;i<=n+1;i++) d[i]=inf;
memset(vis,0,sizeof(vis));
d[x]=0;
priority_queue<pair<int,int> >q;
q.push(make_pair(0,x));
while(q.size())
{
int x=q.top().second;q.pop();
if(vis[x]) continue;
vis[x]=1;
for(int i=0;i<v[x].size();i++)
{
int t=v[x][i].first;
int z=v[x][i].second;
if(d[t]>d[x]+z)
{
d[t]=d[x]+z;
q.push(make_pair(-d[t],t));
}
}
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
if(i-a[i]>=1) v[i-a[i]].push_back(make_pair(i,1));//反向建边
if(i+a[i]<=n) v[i+a[i]].push_back(make_pair(i,1));
}
for(int i=1;i<=n;i++)
{
if(a[i]%2)
{
v[0].push_back(make_pair(i,0));
}else
{
v[n+1].push_back(make_pair(i,0));
}
}
memset(ans,-1,sizeof(ans));
dijkstra(0);
for(int i=1;i<=n;i++)
{
if(a[i]%2==0&&d[i]!=inf) ans[i]=d[i];
}
dijkstra(n+1);
for(int i=1;i<=n;i++)
{
if(a[i]%2&&d[i]!=inf) ans[i]=d[i];
}
for(int i=1;i<=n;i++) printf("%d ",ans[i]);
}