计蒜课 光合作用 百度之星2012.12.23题目1

链接：https://nanti.jisuanke.com/t/38

du熊是个爱学习的孩子，他总喜欢在生活中做一些小实验，这次du熊想研究一下光合作用。

du熊的实验材料有如下几样：神奇的种子，普通的纸箱和一些光源。一开始du熊将种子均匀的种在了箱子底部，你可以将其看成X轴，种子的位置为X轴上的点。然后du熊用纸板将箱子盖住，并在纸板上安装了一些光源（具体见图）。神奇的种子会在有光的情况下一直向上生长直到没光为止。现在du熊想知道当实验结束时每颗种子的高度是多少？

顶上的为光源，光源两边与顶部的夹角都为45°，黄色部分为光照，绿色的为植物。

输入第一行给出一个T，表示测试数据的组数。每组数据的第一行是三个整数n,m,H(1≤n≤100,000, 0≤m≤100,000, 1≤H≤10,000)，n表示种子数(编号1-n)，m表示光源数，H表示箱子的高度。接下来m行，每行一个整数Xi表示第i个光源在顶部的位置。

对于每组测试数据，请输出n行，每行一个数表示第i颗种子的最终高度。

数组没clear导致wa了一天  难过....

很显然 正常人都能想到二分，但是这个题，都是整数的情况下 BFS一趟就OK了 O(n+h)解法..  ，其实 哪怕是小数离散化加优先队列也是可以做的...只不过把大log换成小log了，算常数优化？

...上代码

#include <stdio.h>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
const int maxn = 1e5+1e4+10;
int ans[maxn];
bool vis[maxn];
int n,m,h;
vector<int>a,b;
void init()
{
    memset(ans,-1,sizeof(ans));
    memset(vis,0,sizeof(vis));
    a.clear();
    b.clear();
    scanf("%d%d%d",&n,&m,&h);
}
bool F(int pos)
{
    if(pos>=0&&pos<=n+h&&vis[pos]==false) { return vis[pos] =  true;}
    else return false;
}
void BFS()
{
    int t;
    for(int i=0;i<m;i++)
    {
        scanf("%d",&t);
        vis[t] = true;
        a.push_back(t);
    }
    int lim = a.size(),H = h;
    while(lim&&H)
    {
        for(int i=0;i<lim;i++)
        {
            int x = a[i];
            ans[x] = H;
            if(F(x-1)) b.push_back(x-1);
            if(F(x+1)) b.push_back(x+1);
        }
        swap(a,b);
        H--;
        lim = a.size();
        b.clear();
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    for(int i=0;i<T;i++)
    {
        init();
        BFS();
        for(int i=1;i<=n;i++)
        {
            printf("%d\n",max(ans[i],0));
        }
    }
    return 0;
}

AC代码