HDU 3450 Counting Sequences(树状数组,离散化,中等题)

题意:

            给定一数列,问最多存在多少完美子序列。完美子序列定义如下:

      原序列一共有2n次方个子序列,这些子序列满足这样的关系:子序列元素个数至少2个,且任意相邻的两个元素之差不超过d

 

题解:

每次新来一个元素x,都要去前面扫描有y个大于等于x-d且小于等于x+d的元素,然后以他们为结尾的完美序列数量的和+1(前面都不选),即为新来x结尾的完美序列。对于所有的x线性扫描可得出答案。数量的和当然是用树状数组快速完成。

另外:数大,量小,离散化。

 

 

#include<bits/stdc++.h>

using namespace std;
const int maxn=1e6+100;
const int mod = 9901;
int c[maxn];
int a[maxn],b[maxn];

int lowbit(int x)
{
    return x & -x;
}

int sum(int x)
{
    int res=0;
    while(x>0)
    {
        res=(res+c[x])%mod;
        x -= lowbit(x);
    }
    return res;
}

void add(int x,int v)
{
    while(x<maxn)
    {
        c[x] = (c[x]+v)%mod;
        x += lowbit(x);
    }
}

int main()
{
    int n,d;
    while(cin>>n>>d)
    {   int ans=0;
        memset(c,0,sizeof c);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",a+i);
            b[i] = a[i];
        }
        sort(b+1,b+1+n);
        int m = unique(b+1,b+1+n)-b-1;

        for(int i=1;i<=n;i++)
        {
            int l=lower_bound(b+1,b+1+m,a[i]-d)-b-1;///但可以保证l在范围内呢
            int r=lower_bound(b+1,b+1+m,a[i]+d)-b;///lower_bound 并不一定保证r一定在范围内,但r-1一定在
            int v=lower_bound(b+1,b+1+m,a[i])-b;
            if(r>m||b[r]>a[i]+d)r--;///检查r是否超出范围
            int dp=(sum(r) - sum(l)+mod)%mod;
            add(v,dp+1);
            ans=(ans+dp)%mod;
        }
        cout<<ans<<endl;

    }
}

猜你喜欢

转载自blog.csdn.net/sgh666666/article/details/80443214
今日推荐