hdu 1031 水题

最新推荐文章于 2017-07-21 10:36:00 发布

team79

最新推荐文章于 2017-07-21 10:36:00 发布

阅读量679

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/hzh_0000/article/details/11528307

本文介绍了一个使用 C++ 实现的排序算法案例，该算法通过读取一组数值并计算每组数值的总和，然后对这些总和进行排序，并输出前 K 个最大值对应的原始序列编号。涉及结构体定义、指针操作、快速排序及自定义比较函数等知识点。

AC代码如下：

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

struct node{
	double sum;
	int number;
};

int cmp( const void *a, const void *b ){
	if( ( (struct node*)b )->sum == ( (struct node*)a )->sum ){
		return ( (struct node*)a )->number - ( (struct node*)b )->number;
	}
	return ( (struct node*)b )->sum - ( (struct node*)a )->sum; 
}

int main(){
	int N, M, K;
	struct node s[10000];
	int ans[10000];
	while( scanf( "%d%d%d", &N, &M, &K ) != EOF ){
		memset( s, 0, sizeof( s ) );
		memset( ans, 0, sizeof( ans ) );
		for( int i = 0; i < N; i++ ){
			for( int j = 0; j < M; j++ ){
				double temp;
				cin >> temp;
				s[j].sum += temp;
			}
		}
		for( int i = 0; i < M; i++ ){
			s[i].number = i;
		}
		qsort( s, M, sizeof( struct node ), cmp );
		int temp = 0;
		//while( s[temp].sum == s[temp+K].sum && temp < M - K ){
	//		temp++;
	//	}
		for( int i = temp; i < temp + K; i++ ){
			ans[i] = s[i].number;
		}
		sort( ans, ans + K );
		for( int i = K - 1; i > 0; i-- ){
			cout << ans[i] + 1 << " ";
		}
		cout << ans[0] + 1 << endl;
	}
	return 0;
}