rqnoj350 第k小的数

原题

给出一个长度为N的序列A1,A2,A3,...,AN，其中每项都是

小于10^5的自然数。
现在有M个询问，每个询问都是Ai...Aj中第k小的数等
于多少。
数据范围：
在60%的数据中，1≤N≤1000，1≤M≤1000

在100%的数据中，1≤N≤10000,1≤M≤2000

每次询问的范围都不一样，因此以下的O(n)算法也会在很多个询问的情况下超时：

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
const int maxn=10010;
int A[maxn],B[maxn];

void swap(int *x,int *y)
{
	int t=*x;*x=*y;*y=t;
}

int find(int *A,int x,int y,int k)
{
	if(x==y && k==1) return A[x];
	int p=x,q=y,m=A[x];
	do
	{
		while(p<q && A[q]>=m) q--;
		if(p<q) A[p++]=A[q];
		while(p<q && A[p]<=m) p++;
		if(p<q) A[q--]=A[p];
	}while(p<q);
	A[p]=m;
	//将数组划分为[x,p]和[p+1,y] 
	if(p-x+1>=k) return find(A,x,p,k); //左侧个数比k大，第k小的数在左侧 
	else return find(A,p+1,y,k-(p-x+1));//左侧个数要小于k，则第k小的数出现在右侧，是右侧的第(k-(p-x+1))小的数 
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("in1.txt","r",stdin);
#endif
	int n,m,i;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++) scanf("%d",&A[i]);
	for(i=1;i<=m;i++)
	{
		int k,x,y;
		memcpy(B,A,sizeof(B));
		scanf("%d%d%d",&x,&y,&k);
		printf("%d\n",find(B,x,y,k));
	}
    return 0;
}

正确的做法是，先预处理排序，再统计所求范围的第k小的数：

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
const int maxn=10000+10;
int A[maxn],B[maxn];

void swap(int *x,int *y)
{
	int t=*x;*x=*y;*y=t;
}

void Qsort(int x,int y)
{
	if(x>=y) return;
	int p=x,q=y,m=A[x];
	do
	{
		while(A[p]<m) p++;
		while(A[q]>m) q--;
		if(p<=q)
		{
			swap(&A[p],&A[q]);
			swap(&B[p++],&B[q--]);//下标也交换，A数组保存值，B数组保存下标 
		} 
	}while(p<q);
	Qsort(x,q);
	Qsort(p,y);
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("in1.txt","r",stdin);
#endif
	int n,m,i;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)
	{
		scanf("%d",&A[i]);
		B[i]=i;
	}
	Qsort(1,n);
	for(i=1;i<=m;i++)
	{
		int k,x,y;
		scanf("%d%d%d",&x,&y,&k);
		int t=0;
		for(int j=1;j<=n;j++)
		{
			if(B[j]>=x && B[j]<=y) t++;//判断下标是否在所求范围，如果是统计比k小的数的个数t 
			if(t==k)//找到第k小的数 
			{
				printf("%d\n",A[j])// A[j]即为第k小的数 
				break;
			}	
		}
	}
    return 0;
}