校门外的树4

校门外的树4
题目描述
某校大门外长度为L 的马路上有一排树，每两棵相邻的树之间的间隔都是1 米。我们可
以把马路看成一个数轴，马路的一端在数轴1 的位置，另一端在L 的位置；数轴上的每个整
数点，即1，2，……，L，都种有一棵树。
由于学校要盖楼。所以要将一些比较高的树砍下来当木材。学校想知道砍下某些树之后，
还剩下多少段连续树的树木。
比如马路的长度是5，那么1,2,3,4,5 处各有一颗树。他们的高度分别是2,7,3,5,1。现在
学校想砍掉最高的1 颗树。那么，就会剩下2 段树木，一段是从下标1 开始到下标1 结束的
树，另一段是从下标为3 开始到下标为5 结束的树。
输入格式
输入的第一行有两个整数L（1 <= L <= 106）和M（1 <= M <= 105），L 代表马路的长度，
M 代表总共砍M 次树。
第二行包含L 个不相同的正整数，为树木的高度h（0< h < 108）。
第三行包含M 个正整数k[i]，表示学校想砍掉最高的k[i]颗树(0 < k[i] < 105)。
保证树木不会被砍光
输出格式
总共输出M 行，学校每次砍完一次树(k 颗)，输出一个数字，这个数字占一行，表示剩
下多少段树。
样例输入
5 2
2 7 3 5 1
1 1
样例输出
2
3
提示
成吨的输入量，请进行IO 优化。
本题时间限制3 秒

 

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int i,j,m,n,l,da = 1,vis[1000005],op,t = 0;
struct node
{
    int h;
    int x;
}a[1000005];
int cmp(node a,node b)
{
    return a.h > b.h;
}
int main()
{
    scanf("%d %d",&l,&m);
    for(i = 1;i <= l;i++)
    {
        vis[i] = 1;
        scanf("%d",&a[i].h);
        a[i].x = i;
    }
    sort(a + 1,a + 1 +l,cmp);
    for(i = 1;i <= m;i++)
    {
        scanf("%d",&op);
        for(j = op;j >= 1;j--)
        {
            t = t + 1;
            a[t].h = 0;
            if(vis[a[t].x - 1] == 1 && vis[a[t].x + 1] == 1)
            da++;
            else if(vis[a[t].x - 1] != 1 && vis[a[t].x + 1] != 1)
            da--;
            vis[a[t].x] = 0;
        }
        printf("%d",da);
        if(i != m)
        printf("\n");
    }
    return 0;
}