bzoj 3956: Count

发现好点对互不跨立,所以只有O(n)个,用单调队列找出这O(n)个然后用主席树维护就可以了。
对于a[i]相等的情况的单调队列写错了一些细节。
    
    
    
    
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
 
#define ll long long
#define inf 1e9
#define eps 1e-8
#define md
#define N 300010
#define TR 12000010
using namespace std;
struct QQ { int h,id;} q[N];
struct PP { int x,y;} p[2*N];
int ch[TR][2],sz[TR],a[N],root[N];
int cnt;
bool cmp(PP a,PP b) { return a.x==b.x?a.y<b.y:a.x<b.x;}
 
void insert(int &i,int pre,int l,int r,int x)
{
i=++cnt; ch[i][0]=ch[pre][0]; ch[i][1]=ch[pre][1]; sz[i]=sz[pre]+1;
if (l==r) return;
int mid=(l+r)>>1;
if (x<=mid) insert(ch[i][0],ch[pre][0],l,mid,x);
else insert(ch[i][1],ch[pre][1],mid+1,r,x);
}
 
int query(int a,int b,int l,int r,int ql,int qr)
{
if (ql<=l&&r<=qr) return sz[b]-sz[a];
int mid=(l+r)>>1;
if (qr<=mid) return query(ch[a][0],ch[b][0],l,mid,ql,qr);
if (mid+1<=ql) return query(ch[a][1],ch[b][1],mid+1,r,ql,qr);
return query(ch[a][0],ch[b][0],l,mid,ql,qr)+query(ch[a][1],ch[b][1],mid+1,r,ql,qr);
}
 
int main()
{
int n,m,Ty;
scanf("%d%d%d",&n,&m,&Ty);
for (int i=1;i<=n;i++) scanf("%d",&a[i]);
int top=0,tot=0;
for (int i=1;i<=n;i++)
{
bool had=0;
while (top&&q[top].h<=a[i]) { p[++tot]=(PP){q[top].id,i}; if (q[top].h==a[i]) had=1; top--;}
if (top&&!had) p[++tot]=(PP){q[top].id,i};
q[++top]=(QQ){a[i],i};
}
sort(p+1,p+tot+1,cmp);
int now=0;
for (int i=1;i<=n;i++)
{
root[i]=root[i-1];
while (now<tot&&p[now+1].x==i) { now++; insert(root[i],root[i],1,n,p[now].y);}
}
int ans=0;
for (int i=1;i<=m;i++)
{
int l,r;
scanf("%d%d",&l,&r);
if (Ty) { l=(l+ans-1)%n+1; r=(r+ans-1)%n+1; if (l>r) swap(l,r); }
ans=query(root[l-1],root[r],1,n,l,r);
printf("%d\n",ans);
}
return 0;
}

你可能感兴趣的:(bzoj 3956: Count)