CodeForces 1272E : Nearest Opposite Parity

https://codeforces.com/problemset/problem/1272/E

输入 n(≤2e5) 和长为 n 的数组 a(1≤a[i]≤n)，下标从 1 开始。

从位置 i，你可以移动到位置 i-a[i] 或者 i+a[i]，移动后的位置必须在 [1,n] 内。
定义 d(i) 表示从位置 i 出发，移动到某个位置 j 的最小移动次数，要求 a[i] 和 a[j] 的奇偶性不同。如果不存在这样的 j，则 d(i) 为 -1。输出 d(1),d(2),...,d(n)。

解：建图，i ->i + a[i] 和 i-> i - a[i]. 一开始考虑用dfs，d[i] = min(dfs(i, i + a[i]), dfs(i, i - a[i]))，一种dp思想，但图存在环，有可能出现d[i + a[i]]的最优解依赖于d[i]的最优解，这样就出现了后效性，无法得到局部最优解。

考虑反向建图，以偶数点为起点（所有偶数点的dist为0），用最短路算法，就可以得到所有奇数点的最优解，最短路算法可以不断更新i点的最短路，此题边权为1，可以用bfs。

以奇数点为起点（所有奇数点的dist为0），可以得到所有偶数点的最优解。

#include <iostream>
#include <algorithm>
#include <vector>
#include <functional>
#include <queue>

using namespace std;

int main(){
	int n;
	cin >> n;
	vector<int> a(n + 1);
	vector<vector<int>> grid(n + 1);
	vector<int> dist1(n + 1, n + 1);
	vector<int> dist2(n + 1, n + 1);
	for (int i = 1; i <= n; ++i){
		cin >> a[i];
		if (a[i] & 1) dist2[i] = 0;
		else dist1[i] = 0;
		int u = i + a[i];
		if (u <= n) grid[u].push_back(i);
		u = i - a[i];
		if (u > 0) grid[u].push_back(i);
	}

	
	vector<bool> vis(n + 1);



	bfs(grid, n, dist1);
	bfs(grid, n, dist2);

	for (int i = 1; i <= n; ++i){
		if (dist1[i] > n) dist1[i] = -1;
		if (dist2[i] > n) dist2[i] = -1;
		if (a[i] & 1) cout << dist1[i];
		else cout << dist2[i];
		if (i < n)cout << " ";
	}
	cout << endl;
	return 0;
}