蒜头是个爱学习的孩子,他总喜欢在生活中做一些小实验,这次蒜头想研究一下光合作用。
蒜头的实验材料有如下几样:神奇的种子,普通的纸箱和一些光源。一开始,蒜头将种子均匀的种在了箱子底部,你可以将其看成 X 轴,种子的位置为 X 轴上的点。然后蒜头用纸板将箱子盖住,并在纸板上安装了一些光源(具体见图)。神奇的种子会在有光的情况下一直向上生长直到没光为止。现在蒜头想知道当实验结束时每颗种子的高度是多少?
顶上的为光源,光源两边与顶部的夹角都为 45∘,黄色部分为光照,绿色的为植物。
输入格式
第一行输入一个整数 T,表示测试数据的组数。
每组数据的第一行是三个整数 n,m,H(1≤n,m≤105,0≤m≤105,1≤H≤104),n 表示种子数(编号为 1⋯n),m 表示光源数,H 表示箱子的高度。
接下来 m 行,每行一个整数 Xi 表示第 i 个光源在顶部的位置。
输出格式
对于每组测试数据,请输出 n 行,每行一个数表示第 i 颗种子的最终高度。
样例输入
2 7 1 2 4 4 4 1 1 2 3 4
样例输出
0 0 1 2 1 0 0 1 1 1 1
二分查找植物的前后两个光源,然后判断大小
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<map>
using namespace std;
int main(){
int t;
cin>>t;
while(t--){
int n,m,h;
scanf("%d%d%d",&n,&m,&h);
int num[100000+10]={0};
for(int i=1;i<=m;i++) scanf("%d",&num[i]);
sort(num+1,num+m+1);
for(int i=1;i<=n;i++){
int index=lower_bound(num+1,num+m+1,i)-num;
int maxn=0;
if(m==0){
printf("0\n");
continue;
}
if(index==m+1){
if(i-num[index-1]>=h) maxn=0;
else maxn=h-(i-num[index-1]);
printf("%d\n",maxn);
continue;
}
if(index==1){
if(num[index]-i>=h) maxn=0;
else maxn=h-(num[index]-i);
printf("%d\n",maxn);
continue;
}
if(num[index]==i) maxn=h;
else if(num[index]-i<=h&&num[index]-i<i-num[index-1]) maxn=h-(num[index]-i);
else if(i-num[index-1]<=h&&num[index]-i>=i-num[index-1])maxn=h-(i-num[index-1]);
printf("%d\n",maxn);
}
}
return 0;
}