链接:
https://codeforces.com/problemset/problem/1547/E
题意:
有n个方格,每个方格都有一个气温情况,给你k个气温,a[i]=t,其他格子为min[1≤j≤k](tj+|aj−i|);求每个格子的气温情况。
本题可以先只看一个方向的气温情况,迭代更新每个格子,再看另一个方向,迭代更新格子。最后每个格子取最小值即可。
代码如下:
#include<iostream>
#include<vector>
#include<cmath>
#include<map>
#include<algorithm>
#include<string>
#include<string.h>
#include<random>
using namespace std;
typedef long long ll;
int num[300003];
int r[300003];
int l[300003];
int t[300003];
int main() {
int q;
cin >> q;
while (q--) {
memset(t, 0x3f, sizeof(t));
memset(l, 0x3f, sizeof(l));
memset(r, 0x3f, sizeof(r));
int n,k;
cin >> n>>k;
for (int i = 0; i < k; i++) {
cin >> num[i];
}
for (int i = 0; i < k; i++) {
cin >> t[num[i]];
}
for (int i = 1; i <= n; i++) {
l[i] = min(l[i - 1] + 1, t[i]);
}
for (int i = n; i >= 1; i--) {
r[i] = min(r[i + 1] + 1, t[i]);
}
for (int i = 1; i <= n; i++) {
cout << min(r[i], l[i]) << " ";
}
cout << endl;
}
return 0;
}