思路:首先如果都是-1的话,我们可以输出1 2循环,否则。
首先处理首尾的连续-1串,这个也很好处理,最后我们处理两个非-1之间的-1串。
对于左边的数a[l]和右边的数a[r]:
如果a[l] > a[r],那么就处理a[l + 1],让左边的数变小,否则处理a[r - 1]。
代码:
void solve() {
int n;
std::cin >> n;
std::vector<int> a(n);
for (int i = 0; i < n; i++) {
std::cin >> a[i];
}
if (std::count(a.begin(), a.end(), -1) == n) {
for (int i = 0; i < n; i++) {
a[i] = i % 2 + 1;
}
} else {
for (int i = 0, j = -1; i <= n; i++) {
if (i == n || a[i] != -1) {
if (j == -1) {
for (int k = i - 1; k >= 0; k--) {
a[k] = a[k + 1] == 1 ? 2 : a[k + 1] / 2;
}
} else if (i == n) {
for (int k = j + 1; k < n; k++) {
a[k] = a[k - 1] == 1 ? 2 : a[k - 1] / 2;
}
} else {
int l = j, r = i;
while (l + 1 < r) {
if (a[l] > a[r]) {
a[l + 1] = a[l] == 1 ? 2 : a[l] / 2;
l++;
} else {
a[r - 1] = a[r] == 1 ? 2 : a[r] / 2;
r--;
}
}
if (a[l] != a[r] / 2 && a[r] != a[l] / 2) {
std::cout << -1 << "\n";
return;
}
}
j = i;
}
}
}
for (int i = 0; i < n; i++) {
std::cout << a[i] << " \n"[i == n - 1];
}
}