杭电1031题及C++中sort的用法



上链接：杭电1031

题目大意：
给出N个人对M个元素的满意度，需要添加K个元素到T-shirt上，使得满意度最大

两次排序便可以解决
用c++自带的排序函数sort很方便

#include<iostream>
#include<algorithm>
using namespace std;

struct element{
	float sum;
	int num;
};
int cmp1(element &a,element &b)
{
	if(a.sum>b.sum)
	   return 1;
	else if(a.sum==b.sum)
	     {
	     	if(a.num<b.num)
	     	   return 1;
	     	else
	     	   return 0;
		
	     }
	     else
	        return 0;
}
int cmp2(int &a,int &b)
{
	return a>b;
}
	float a[500][500];
	int out[500];
	element b[500];
int main()
{
	int N,M,K; 
  while(scanf("%d %d %d",&N,&M,&K)!=EOF)
    {


	int i,j;

	for(i=0;i<N;i++)	
	   for(j=0;j<M;j++)
	      cin>>a[i][j];
     
     for(j=0;j<M;j++)       //
     {
     	b[j].num=j+1;
     	b[j].sum=0;
     	for(i=0;i<N;i++)
     	 {
     	 	b[j].sum=b[j].sum+a[i][j];
		  }
	 }
    sort(b,b+M,cmp1);
	
	for(i=0;i<K;i++)
	    out[i]=b[i].num;
	sort(out,out+K,cmp2);
	for(i=0;i<K;i++)
	{
		if(i!=K-1)
		   cout<<out[i]<<" ";
		 else
		    cout<<out[i]<<endl; 
		
		 }	 

  }

	
	
}

下面是关于sort函数的用法

-------------------------------------------------------------------------------------

头文件：
#include <algorithm>
using namespace std;

1.默认的sort函数是按升序排。对应于1）
sort(a,a+n); //两个参数分别为待排序数组的首地址和尾地址
2.可以自己写一个cmp函数，按特定意图进行排序。对应于2）
例如：
int cmp( const int &a, const int &b ){
if( a > b )
return 1;
else
return 0;
}
sort(a,a+n,cmp);
是对数组a降序排序
又如：
int cmp( const POINT &a, const POINT &b ){
if( a.x < b.x )
return 1;
else
if( a.x == b.x ){
if( a.y < b.y )
return 1;
else
return 0;
}
else
return 0;
}
sort(a,a+n,cmp);
是先按x升序排序，若x值相等则按y升序排

与此类似的还有C中的qsort，以下同附上qsort的使用方法：

#include <stdlib.h>

格式 qsort(array_name,data_number,sizeof(data_type),compare_function_name) (void*)bsearch (pointer_to_key_word,array_name,find_number,

sizeof(data_type),compare_function_name)

e.g.

int Cmp(const void*a,const void *b)

{

int*pa=(int*)a,*pb=(int*)b;

if(*pa>*pb) return 1;

else if (*pa==*pb) return 0;

else return -1;

}

qsort(data,N,sizeof(int),Cmp); // 对int型数组进行快速排序（非降序排列）

p=(int*)bsearch(&a,data,n,sizeof(int),Cmp);