杂题 区间第K大(kth)
问题 C: 区间第K大(kth)
时间限制: 20 Sec 内存限制: 256 MB提交: 48 解决: 37
[提交][状态][讨论版]
题目描述
想必大家对区间第K大问题相当熟悉了。这个问题是这样的,给一串序列和若干询问,每个询问查询某段连续区间中第K大的数。现在我们考虑一个该问题的“Reverse”版本。现在我们给出序列和Q个询问,每个询问给出K_i和X_i,询问有多少区间其第K_i大的数为X_i。
输入
第一行一个整数N和Q,表示序列长度和询问个数。
第二行N个整数A_i,表示序列中的元素。
接下来Q行,每行两个整数K_i和X_i,表示询问。
输出
一共Q行,每行表示对应询问的答案。
样例输入
3 21 1 21 12 1
样例输出
33
提示
对于20%的数据,N<=100
对于40%的数据,N<=500
对于100%的数据,N<=2000,Q<=2000000, 1 ≤ K_i ≤ N,1 ≤ X_i ≤ N
正解是N^2预处理每个点。shu[i]表示到第i位时,有多少个数比当前枚举的点大。sum1[i]记录这个点之前包含i个比他大的点的区间数。sum2[i]就是后面的。
f[a[i]][k+j+1]+=sum1[j]*sum2[k],这个就很好解释了,a[i]为第k+j+1大时就是他前面有j个比他大的区间数*后面有k个比他大的区间数。。。
注意细节,因为重复出现的值不去重,所以。。前后中有一个判断是不带=的,这样才能对上。
sum1[0]=sum2[0]=1,包含他自己。
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<algorithm> using namespace std; int read() { int sum=0,f=1;char x=getchar(); while(x<'0'||x>'9'){if(x=='-')f=-1;x=getchar();} while(x>='0'&&x<='9'){sum=(sum<<1)+(sum<<3)+x-'0';x=getchar();} return sum*f; } int n,q,a[2005],f[2005][2005],shu[2005],num1[2005],num2[2005]; int main() { freopen("kth.in","r",stdin); freopen("kth.out","w",stdout); n=read();q=read(); for(int i=1;i<=n;i++)a[i]=read(); for(int i=1;i<=n;i++) { memset(shu,0,sizeof(shu)); memset(num1,0,sizeof(num1)); memset(num2,0,sizeof(num2)); //shu[i]=1; for(int j=i-1;j>=1;j--) { shu[j]=shu[j+1]; if(a[j]>a[i])shu[j]++; num1[shu[j]]++; } for(int j=i+1;j<=n;j++) { shu[j]=shu[j-1]; if(a[j]>=a[i])shu[j]++; num2[shu[j]]++; }num2[0]++;num1[0]++; for(int j=0;j<=shu[1];j++) for(int k=0;k<=shu[n];k++) f[a[i]][k+j+1]+=num2[k]*num1[j]; // for(int j=0;j<=n;j++)cout<<num1[j]<<" "; // cout<<endl; // for(int j=0;j<=n;j++)cout<<num2[j]<<" "; // cout<<endl; } int x,y; for(int i=1;i<=q;i++) { x=read(); y=read(); printf("%d\n",f[y][x]); } }