传送门
思维。先给定了若干空调的位置a和温度t,需要推算出所有位置空调的温度。
表达式就是最小温度加其与i距离。
常规超时,所以递推。
若已经知道b处的a和t,则与b距离为1的格子为:b温度+距离1
因为表达式min,从左到右和从右到左都递推一遍,维护最小值便可。
#include<bits/stdc++.h>
using namespace std;
#define N 300005
#define inf 0x3f3f3f3f
int a[N],b[N];
int main()
{
int r,n,k;
scanf("%d",&r);
getchar();
while(r--)
{
getchar();
memset(b,inf,sizeof(b));
scanf("%d%d",&n,&k);
for(int i=1;i<=k;i++) scanf("%d",&a[i]);
for(int i=1;i<=k;i++)
{
scanf("%d",&b[a[i]]);
}
for(int i=1;i<=n;i++) b[i]=min(b[i],b[i-1]+1);
for(int i=n;i>=1;i--) b[i]=min(b[i],b[i+1]+1);
for(int i=1;i<=n;i++)
{
printf("%d",b[i]);
if(i==n)printf("\n");
else printf(" ");
}
}
}