https://www.nowcoder.com/acm/contest/124#question

題意 找第一個不小於K的數的下標，然後對它前一個數加一

解析 我們可以維護一個最大值數組 1到 i的 最大值 就是max[ i ] 二分找到最左邊的值 但是 找到的前一個加1 要用線段樹來維護最大值

但是 這麽寫會超時。。。

超時代碼

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1000000007,maxn=1e6+50;
int sum[maxn<<2];
int a[maxn],n,m;
 
void PushUP(int rt)
{
    sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
void Build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=a[l];
        return;

    }
    int m=(l+r)>>1;
    Build(l,m,rt<<1);
    Build(m+1,r,rt<<1|1);
    PushUP(rt);
}
void Update(int L,int
 
 C,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]+=C;
        return;
    }
    int m=(l+r)>>1;
    if(L<=m)
        Update(L,C,l,m,rt<<1);
    else
        Update(L,C,m+1,r,rt<<1|1);
    PushUP(rt);
}
int Query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        
 
return sum[rt];
    }
    int m=(l+r)>>1;
    int ans=-100000000;
    if(L<=m)
        ans=max(ans,Query(L,R,l,m,rt<<1));
    if(R>m)
        ans=max(ans,Query(L,R,m+1,r,rt<<1|1));
    return ans;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(sum,0,sizeof(sum));
        memset(a,0,sizeof(a));
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
        }
        Build(1,n,1);
        int l,r,k;
        while(m--)
        {
            l=1,r=n;
            scanf("%d",&k);
            //cout<<Query(1,n,1,n,1)<<endl;
            if(Query(1,n,1,n,1)<k)
            {
                printf("are you ok\n");
                continue;
            }
            while(l<=r)
            {
                int mid=(l+r)>>1;
               // cout<<mid<<" "<<Query(1,mid,1,n,1)<<endl;
                if(Query(1,mid,1,n,1)>=k)
                    r=mid-1;
                else
                    l=mid+1;
            }
            printf("%d\n",l-1);
            if(l-1)
                Update(l-1,1,1,n,1);
        }
    }
}

q的 詢問比較多應該是卡了常數 我們要優化一下 因為 線段樹查詢的時候就是二分 區間最大值是遞增的 我們直接利用這個特點來操作

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1000000007,maxn=1e6+50;
int sum[maxn<<2];
int a[maxn],n,m;
void PushUP(int rt)
{
    sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
void Build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=a[l];
        return;

    }
    int m=(l+r)>>1;
    Build(l,m,rt<<1);
    Build(m+1,r,rt<<1|1);
    PushUP(rt);
}
void Update(int L,int C,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]+=C;
        return;
    }
    int m=(l+r)>>1;
    if(L<=m)
        Update(L,C,l,m,rt<<1);
    else
        Update(L,C,m+1,r,rt<<1|1);
    PushUP(rt);
}
int query(int L,int R,int l,int r,int rt,int p)
{
    if(l==r)
    {
        return l;
    }
    int m=(l+r)>>1;
    if(sum[rt<<1]>=p)                    //二分查詢
        return query(L,R,l,m,rt<<1,p);
    return query(L,R,m+1,r,rt<<1|1,p);
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(sum,0,sizeof(sum));
        memset(a,0,sizeof(a));
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
        }
        Build(1,n,1);
        int l,r,k;
        while(m--)
        {
            l=1,r=n;
            scanf("%d",&k);
            if(sum[1]<k)
            {
                printf("are you ok\n");
                continue;
            }
            int ans=query(1,n,1,n,1,k);
            printf("%d\n",ans-1);
            if(ans-1)
                Update(ans-1,1,1,n,1);
        }
    }
}

其實 還有更簡單的做法 因為 修改的是前一個值 而且找的是滿足條件中最左邊的 所以前一個+1 並不會 影響數組的單調性 變得只有前面一個的最大值 更新一下就好了

#include<bits/stdc++.h>
using namespace std;
int a[1000005],b[1000005];
int n,q,k;
int main()
{
    while(~scanf("%d%d",&n,&q))
    {
        int mx=0;
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
            b[i]=mx=max(mx,a[i]);
        }
        while(q--)
        {
            scanf("%d",&k);
            int l=lower_bound(b,b+n,k)-b;
            if(l==n){
                printf("are you ok\n");
                continue;
            }
            printf("%d\n",l);
            if(l==0)continue;
            a[l-1]++;
            b[l-1]=max(a[l-1],b[l-1]);
        }
    }
}

牛客網 中南林業科技大學第十一屆程序設計大賽J題 二分+線段樹