标签:
6 2 3 2 1 4 3 5 3 4
Case #1: 2 2
思路:满足条件的区间必然是 (区间内的最大值-最小值=len-1)&&(区间内的不同值==len)
第一个条件可以用RMQ解决,第二个条件滑窗dp
#include<cstdio> #include<iostream> #include<cstring> #include<algorithm> #include<map> using namespace std; const int N = 1e4+100; int num[N]; int n,m; int mm[N]; int maxn[30][N]; int minn[30][N]; void initRMQ(int n) { for(int i=1;i<=n;i++) maxn[0][i]=num[i],minn[0][i]=num[i]; for(int i=1;i<=mm[n];i++) for(int j=1;j+(1<<i)-1<=n;j++) { int a=maxn[i-1][j]; int b=maxn[i-1][j+(1<<(i-1))]; if(a>b) maxn[i][j]=a; else maxn[i][j]=b; a=minn[i-1][j],b=minn[i-1][j+(1<<(i-1))]; if(a<b) minn[i][j]=a; else minn[i][j]=b; } } int askmax(int a,int b) { int t; t=mm[b-a+1]; b-=(1<<t)-1; a=maxn[t][a]; b=maxn[t][b]; return a>b ? a:b; } int askmin(int a,int b) { int t; t=mm[b-a+1]; b-=(1<<t)-1; a=minn[t][a]; b=minn[t][b]; return a<b ? a:b; } int cnt[N]; struct node { int v,id; }hs[N]; int mp[N]; int top; void ini() { mm[0]=-1; for(int i=1;i<=n;i++) mm[i]=((i&(i-1))==0)? mm[i-1]+1:mm[i-1]; top=0; } bool cmp(const node &a ,const node &b) { return a.v<b.v; } int cas; int main() { while(~scanf("%d%d",&n,&m)) { ini(); printf("Case #%d:\n",++cas); for(int i=1;i<=n;i++) scanf("%d",&num[i]),hs[i].v=num[i],hs[i].id=i; sort(hs+1,hs+1+n,cmp); mp[hs[1].id]=++top; for(int i=2;i<=n;i++) if(hs[i].v!=hs[i-1].v)mp[hs[i].id]=++top;else mp[hs[i].id]=top; initRMQ(n); for(int i=1;i<=m;i++) { int k,c=0; long long ans=0; scanf("%d",&k); if(k>n) {puts("0");continue;} for(int i=1;i<=top;i++) cnt[i]=0; for(int i=1;i<=k;i++) if(cnt[mp[i]]==0) c++,cnt[mp[i]]++; if(askmax(1,k)-askmin(1,k)==k-1&&c==k) ans++; for(int p=2;p+k-1<=n;p++) { if(--cnt[mp[p-1]]==0) c--; if(cnt[mp[p+k-1]]++==0) c++; if(askmax(p,k+p-1)-askmin(p,k+p-1)==k-1&&c==k) ans++; } printf("%I64d\n",ans); } } return 0; }
标签:
原文地址:http://blog.csdn.net/kalilili/article/details/46287485