欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

砍树(贪心)

程序员文章站 2024-03-23 17:58:46
...

1.题目

为了在农场的一块土地上种奶牛吃的草,FJ必须要把前面的N棵树砍掉,这些树紧密地排成一条直线,并且用l~N编号标识,每一棵树都有自己的高度Hi(1≤Hi≤10000)。
FJ想用一种烈性炸药来摧毁这些树,这种烈性炸药除了能摧毁安装了炸药的那棵树以外还能传递压倒两边矮于这一棵树的所有邻近树,直到遇到一棵不低于这一棵树的树为止。
例如:一排树的高度如下:1 2 5 4 3 3 6 6 2,如果FJ在第三棵树上装炸药(高度为5),那么第二棵树也同样给压倒(高度为2<5),第一棵树也同样倒下(高度为1<2),再来看另一边第四棵树(高度为4<5)和第五棵树(高度为3<4)同样也给压倒。剩下的状态为:* * * * * 3 6 6 2,接下来在第7和第8棵树上安装炸药就可以把剩下的树毁掉。
请你帮助Fj利用最少的炸药把这些树毁掉。

输入
第1行:一个整数N(1≤N≤50000);
第2~N+1行:包含各棵树的高度Hi。
输出
l~?行:每一行为一个整数,代表安装炸药的树的编号,按照升序输出。

  • 样例输入
    9
    1
    2
    5
    4
    3
    3
    6
    6
    2
  • 样例输出
    3
    7
    8

2.思路

想法起始很简单,我的想法是:每次找到一个最大的数(允许它后面的数跟它一样),然后前面肯定都没了,后面再找,只要存在a[i]<=a[i+1],就说明这个炸弹已经用完了,这样的想法一直找下去,找到最后一个就得了

3.WA掉的代码

  • 正所谓:想要加入acm,得先去wam逛一圈。wa掉的代码如下
#include <iostream>
#include<bits/stdc++.h>
using namespace std;
int a[500000],n,b[500000],cnt,k;
int main()
{
    cin>>n;
    int i;
    for(i=0;i<n;i++)cin>>a[i];
    i=0;
    for(;i<n-1;i++)
    {
        if(a[i]>=a[i+1])
        {
            b[cnt++]=i+1;
            for(;i<n-1;i++)
            {
                if(a[i]<=a[i+1])
                {
                    break;
                }
            }
        }
    }
    for(i=0;i<cnt;i++)
    {
        cout<<b[i]<<endl;
    }
    return 0;
}

自我感觉良好,但是wa掉了,经过思考,换了一组样例,将n=10,在原有样例的后面又加了一个2,但是结果仍然是3 7 8,那么问题得到解决,遍历的过程中没有有效地对最后一个数据进行处理

4.AC代码

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
int a[500000],n,b[500000],cnt,k;
int main()
{
    cin>>n;
    int i;
    for(i=0;i<n;i++)cin>>a[i];
    i=0;
    for(;i<n-1;i++)
    {
        if(a[i]>=a[i+1])
        {
            b[cnt++]=i+1;
            for(;i<n-1;i++)
            {
                if(a[i]<=a[i+1])
                {
                    break;
                }
            }
        }
    }
    if(a[n-2]<=a[n-1])b[cnt++]=n;
    for(i=0;i<cnt;i++)
    {
        cout<<b[i]<<endl;
    }
    return 0;
}

5.代码优化

同样的思想,代码优化为下述:

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n,a[50005],i;
    cin>>n;
    for(i=0;i<n;i++)cin>>a[i];
    for(i=0;i<n;i++)
    {
        for(;a[i]<a[i+1];i++);
        cout<<i+1<<endl;
        for(;a[i]>a[i+1];i++);
    }
    return 0;
}
相关标签: 砍树(贪心)