第K个数(快速排序)

题目描述 Description

      给定一个长度为N(0<n<=10000)的序列，保证每一个序列中的数字a[i]是小于maxlongint的非负整数 ，编程要求求出整个序列中第k大的数字减去第k小的数字的值m，并判断m是否为质数。(0<k<=n)

输入描述 Input Description

      第一行为2个数n，k（含义如上题）
      第二行为n个数，表示这个序列

输出描述 Output Description

      如果m为质数则
      第一行为'YES'(没有引号）
      第二行为这个数m
      否则 
      第一行为'NO'
      第二行为这个数m

样例输入 Sample Input

5 2
1 2 3 4 5

样例输出 Sample Output

YES
2

数据范围及提示 Data Size & Hint

      20%数据满足0<n<=10
      50%数据满足0<n<=5000
      100%数据满足0<n<=10000 

这题考察的还是排序和质数判断，唯一要注意的是此题需要排序的数目比较多，用一般的排序方法可能会超时，所以用的快速排序，还有一点声明数组时要用long long int类型

#include <stdio.h>
#define M 10000
//声明函数
void sort(long long int *num,int begin,int end);
void swap(long long int *a,long long int *b);
int judge(long long int num);
int main(void)
{
	long long int num[M]={0};
	long long int n=0,k=0;
	long long int i=0,tmp=0;
	scanf("%lld %lld",&n,&k);
	for(i=0;i<n;i++)
	{
		scanf("%lld",num+i);
	}
	sort(num,0,n-1);
	tmp = num[n-k]-num[k-1];
	if(judge(tmp)) puts("YES");
	else puts("NO");
	printf("%lld\n",tmp);
	return 0;
}
void swap(long long int *a,long long int *b)
{
	long long int tmp = *a;
	*a = *b;
	*b = tmp;
}
//判断num是否为质数，注意遍历的i也应该为long long int型
int judge(long long int num)
{
	if(num<2) return 0;
	long long int i=0;
	for(i=2;i*i<num;i++)
	{
		if(num%i==0) return 0;
	}
	if(i*i==num) return 0;
	return 1;
}
//快速排序
void sort(long long int *num,int begin,int end)
{
	int i=0,j=0;
	if(begin<end)
	{
		i = begin+1;
		j = end;
		while(i<j){
			if(num[begin]<num[i])
			{
				swap(num+i,num+j);
				j--;
			}
			else i++;
		}
		if(num[begin]<=num[i])
		{
			i--;
		}
		swap(num+begin,num+i);
		sort(num,begin,i);
		sort(num,j,end);
	}
}