首页 > 代码库 > 计蒜课 光合作用 百度之星2012.12.23题目1

计蒜课 光合作用 百度之星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代码

 

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