CSUOJ2078-查找第k大(读入挂)

查找第k大

Submit Page    Summary    Time Limit: 1 Sec     Memory Limit: 128 Mb     Submitted: 316     Solved: 20    

---

Description

小W有很强的好胜心，也有很明确的目标，总是希望当第k名，但是小W太菜了，经常达不到目标，于是他每次考试后都想知道第k名的分数是多少，然后以它为目标。 现在给出了每个人的分数，请求编程能力很强的你帮他迅速找到第k名的分数为多少，这样他才有更多的时间去学习。

Input

第一行为一个正整数t代表有t组数据。每组数据第一行为两个正整数n和k，第二行为n个正整数。 1 < =k < =n < =107

Output

对于每组数据，输出第k大的数

Sample Input

1
6 2
1 2 3 4 5 6

Sample Output

5

Hint

#include<cstdio> 
#include<algorithm> 
namespace fastIO 
{ 
	#define BUF_SIZE 100000 	
	bool IOerror = 0; 	
	inline char nc() 	
	{ 		
		static char buf[BUF_SIZE], *p1 = buf + BUF_SIZE, *pend = buf + BUF_SIZE;
		 	if ( p1 == pend ) 		
			 {
			  		p1	= buf; 			
					pend	= buf + fread( buf, 1, BUF_SIZE, stdin ); 			
					if ( pend == p1 ) 			
					{ 		
							IOerror = 1; 	
							return(-1); 
					} 		
			} 		
		
			return(*p1++); 	
		
	} 	
	inline bool blank( char ch ) 	
	{ 		
		return(ch == ' ' || ch == '
' || ch == '
' || ch == '	'); 	
	} 	
	inline void rd( int &x ) 	
	{ 		
		char ch; 		
		while ( blank( ch = nc() ) ) 			
		; 		
		if ( IOerror ) 			
		return; 		
		for ( x = ch - '0'; (ch = nc() ) >= '0' && ch <= '9'; x = x * 10 + ch - '0' ) 			
		; 
	} 
	#undef BUF_SIZE 
}; 
using namespace fastIO; 
using namespace std; 
const int N = 1e7+10; 
int arrNum[N];
int Partition(int *arrNum, int l, int r)
{       
    if(arrNum == NULL || l > r)
	{  
        return -1;  
    }  
    int mid = (l+r)>>1;  
    swap(arrNum[mid], arrNum[r]);  
    int leftSeqIndex = l;  
    
    for(int i = l; i < r; i++)
	{  
        if(arrNum[i] < arrNum[r])
		{  
           if(i > leftSeqIndex)
		   {  
                swap(arrNum[i], arrNum[leftSeqIndex]);  
           }  
           ++leftSeqIndex;  
        }  
    }   
      
    swap(arrNum[leftSeqIndex], arrNum[r]);  
    return leftSeqIndex;   
}  
  
 
int GetNumber(int *arrNum, int n, int k)
{   
    if(arrNum == NULL || n <= 0 || k <= 0 || k > n)
	{  
        return -1;  
    }  
    
    k = n-k+1;  
    int l = 0;  
    int r = n-1;  
    while(l <= r)
	{  
        int index = Partition(arrNum, l, r);  
        if(index+1 == k)
		{    
            return arrNum[index];  
        }  
        else if(index+1 < k)
		{   
            l = index+1;   
        }  
        else
		{  
            r = index-1;  
        }  
    }  
}   
 
int main() 
{ 	
	int t,n,k; 	
	rd(t);
	while(t--)
	{ 		
		rd(n); 	
		rd(k); 		
		for(int i=0;i<n;++i)
		{ 			
			rd(arrNum[i]); 		
		} 
		int value = GetNumber(arrNum, n, k);	
		 	printf("%d
",value); 	
	} 
	return 0;
} 


/**********************************************************************
	Problem: 2078
	User: song_hai_lei
	Language: C++
	Result: AC
	Time:924 ms
	Memory:40280 kb
**********************************************************************/