hdu2836（2009多校第三场） 相邻项绝对值小于等于k的序列数（线段树+sort unique lowerbound）...

和codeforces上次那个题目差不多吧，都是用线段树求和加速dp

为了下次好找，把几个stl放标题哈哈哈

 1 #include<stdio.h>
 2 #include<string.h>
 3 #include<algorithm>
 4 using namespace std;
 5 #define MOD 9901
 6 int sumv[400005],a[100005],b[100005];
 7 int query(int o,int l,int r,int ql,int qr)
 8 {
 9   int mid=l+(r-l)/2,ans=0;
10   if (ql<=l&&qr>=r) return sumv[o];
11   if (ql<=mid) ans+=query(o*2,l,mid,ql,qr);
12   if (qr>mid) ans+=query(o*2+1,mid+1,r,ql,qr);
13   return ans%MOD;
14 }
15 void update(int o,int l,int r,int p,int v)
16 {
17   int mid=l+(r-l)/2;
18   if (l==r) sumv[o]=(sumv[o]+v)%MOD;
19   else{
20     if (p<=mid) update(o*2,l,mid,p,v);
21     if (p>mid) update(o*2+1,mid+1,r,p,v);
22     sumv[o]=(sumv[o*2]+sumv[o*2+1])%MOD;
23   }
24   return;
25 }
26 int main()
27 {
28   int n,k,i,cnt,t1,t2,tmp,ans;
29   while (~scanf("%d%d",&n,&k))
30   {
31     for (i=1;i<=n;i++)
32     {
33       scanf("%d",&a[i]);
34       b[i]=a[i];
35     }
36     sort(b+1,b+n+1);
37     cnt=unique(b+1,b+n+1)-b-1;
38     memset(sumv,0,sizeof(sumv));
39     ans=0;
40     for (i=1;i<=n;i++)
41     {
42       t1=lower_bound(b+1,b+cnt+1,a[i]-k)-b;
43       t2=upper_bound(b+1,b+cnt+1,a[i]+k)-b-1;
44       tmp=query(1,1,cnt,t1,t2);
45       ans=(ans+tmp)%MOD;
46       t1=lower_bound(b+1,b+cnt+1,a[i])-b;
47       update(1,1,cnt,t1,tmp+1);
48     }
49     printf("%d\n",ans);
50   }
51   return 0;
52 }

View Code

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2836

转载于:https://www.cnblogs.com/xiao-xin/articles/4160090.html