1547 E Air Conditioners

1500分

题意

长为n的序列，有k个空调在序列之中，每个空调占一格，每个格子都有一个温度，温度取决于公式，求每一个格子的温度多少

解析

• 设置$c[i]$代表i点的空调的温度，如果没有空调温度为INF
• $le[N],ri[N]$代表只考虑自身及左边，自身及右边，此处的温度（最小值）。

1. 根据dp递推思想，维护数组$ri$只考虑每个点只受左边的空调影响，假设一个点的温度是$t[i]$，$t[i+1],t[i+2]$如果不受处在$i+1,i+2$位置的空调影响，就是$t[i+1]=t[i]+1，t[i+2]=t[i]+2$，但是其可以受自身位置的这个空调影响，因此我们和其比较取一个最小值即可。

2. 我们用同样的思想处理右边

3. 最后遍历数组，对这个点的左右取一个min即可

代码

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+5,inf=0x3f3f3f3f;
struct node{
    int lo,te;
}r[N];
int c[N];//c[i]代表i点的空调的温度，如果没有空调温度为INF
int le[N],ri[N];//代表只考虑自身及左边，自身及右边，此处的温度。
void solve(){
    int n,k;
    cin>>n>>k;
    memset(le,0x3f,sizeof le);
    memset(ri,0x3f,sizeof ri);
    memset(c,0x3f,sizeof c);
    for(int i=0;i<k;i++){
        cin>>r[i].lo;
    }
    for(int i=0;i<k;i++){
        cin>>r[i].te;
        c[r[i].lo]=r[i].te;//转换一下
    }
    //维护一下左边
    for(int i=1;i<=n;i++){
        le[i]=min(le[i-1]+1,c[i]);
    }
    //维护一下右边
    for(int i=n;i>=0;i--){
        ri[i]=min(ri[i+1]+1,c[i]);
    }
 
    for(int i=1;i<=n;i++){
        cout<<min(ri[i],le[i])<<" ";
    }
    cout<<endl;
}
int main(){
    int t ;
    cin>>t;
    while(t--){
        solve();
    }
}