CodeForces 1272E Nearest Opposite Parity(多源bfs)

题目链接

题目大意

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

解题思路

  一个数字可以移动到i+arr[i]和i-arr[i]，设\(d[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\n":"%d ", dis[i]==INF?-1:dis[i]);
    return 0;  
}