HDU 3450(树状数组)

最新推荐文章于 2019-06-29 12:27:16 发布

原创最新推荐文章于 2019-06-29 12:27:16 发布 · 263 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#HDU 3450

数据结构专栏收录该内容

9 篇文章

订阅专栏

本文解析了HDU3450题目，该题旨在求解长度符合特定条件且相邻元素差值不超过给定阈值d的子序列数量。通过使用C++实现树状数组来高效完成查询与更新操作，有效解决了这一问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接:HDU 3450

题目大意:

求长度大于等于且相邻差值不超过d的子序列的个数.

分析:

没什么好说的水题,就是没给a的范围真的天坑.

以下是代码:

#include <set>
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <string>
#include <math.h>
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>

#define Fi first
#define Se second
#define ll long long
#define inf 0x3f3f3f3f
#define lowbit(x) (x&-x)
#define PLL pair<ll,ll>
#define PII pair<int,int>
#define in(b) scanf("%d",&b)
#define l_inf 0x3f3f3f3f3f3f3f
#define mmin(a,b,c) min(a,min(b,c))
#define mmax(a,b,c) max(a,max(b,c))
#define debug(a) cout<<#a<<"="<<a<<endl;
#define debug2(a,b) cout<<#a<<"="<<a<<" "<<#b<<"="<<b<<endl;
using namespace std;
const int N=300010;
const int mod=9901;
int s[N];
void update(int p,int val)
{
    while(p<N)
    {
        s[p]+=val;
        s[p]%=mod;
        p+=lowbit(p);
    }
}
int query(int p)
{
    int ans=0;
    while(p)
    {
        ans+=s[p];
        ans%=mod;
        p-=lowbit(p);
    }
    return ans;
}
int a[N];
vector<int> vc;
int main()
{
    int n,d;
    while(~scanf("%d%d",&n,&d))
    {
        int ans=0;
        memset(s,0,sizeof s);
        vc.clear();
        for(int i=1;i<=n;i++)
        {
            in(a[i]);
            vc.push_back(a[i]);
            vc.push_back(a[i]+d);
            vc.push_back(a[i]-d);
            vc.push_back(a[i]-d-1);
        }
        sort(vc.begin(),vc.end());
        int m=unique(vc.begin(),vc.end())-vc.begin();

        for(int i=1; i<=n;i++)
        {
            int p=lower_bound(vc.begin(),vc.begin()+m,a[i])-vc.begin()+1;
            int r=lower_bound(vc.begin(),vc.begin()+m,a[i]+d)-vc.begin()+1;
            int l=lower_bound(vc.begin(),vc.begin()+m,a[i]-d-1)-vc.begin()+1;
            int ls=query(l);
            int rs=query(r);
            ans=(ans+rs-ls+mod)%mod;
            update(p,(rs-ls+1+mod)%mod);
        }
        printf("%d\n",ans);
    }

    return 0;
}
/*


*/