CodeForces 1272E Nearest Opposite Parity(bfs)

题目链接

题目大意

  一个数字可以移动到i+arr[i]和i-arri，问每个位置移动到与当前的值奇偶性不同的位置需要的次数。

解题思路

  ;一个数字可以移动到i+arr[i]和i-arr[i]，设(d[])表示能到达奇偶性不同的点需要的最小次数，如果这个数一次就可以，那么d[i]=1, 否则，假设我们知道d[i+arr[i]]与d[i-arr[i]]的值，那么d[i] = min(d[i+arr[i]],d[i-arr[i]])+1，从i+arr[i]和i-arr[i]建一条指向i的边，然后以所有d[i]=1的点做起始点开始bfs。

代码

const int maxn = 2e5+10;
const int maxm = 2e6+10;
int n, dis[maxn], arr[maxn], vis[maxn];
vector<int> e[maxn];
int main() {
    cin >> n;
    clr(dis, 0x3f);
    for (int i = 1; i<=n; ++i) {
        scanf("%d", &arr[i]);
    }
    for (int i = 1; i<=n; ++i) {
        if (i-arr[i]>=1) { 
            if ((arr[i]^arr[i-arr[i]])&1) dis[i] = 1;
            e[i-arr[i]].push_back(i);
        }
        if (i+arr[i]<=n) {
            if ((arr[i]^arr[i+arr[i]])&1) dis[i] = 1;
            e[i+arr[i]].push_back(i); 
        }
    }
    queue<int> q;
    for (int i = 1; i<=n; ++i)
        if (dis[i]==1) q.push(i);;
    while(!q.empty()) {
        int u = q.front(); q.pop();
        for (auto v : e[u])
            if (dis[v]>dis[u]+1) dis[v]=dis[u]+1, q.push(v);
    }
    for (int i = 1; i<=n; ++i) printf(i==n?"%d
":"%d ", dis[i]==INF?-1:dis[i]);
    return 0;  
}