hdu_1031_Design T-Shirt(模拟)

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1031

题意：输入n,m,k，n表示有多少个人，m表示共有多少个设计元素，k表示他要选择的设计元素的个数；

n个人都对m中设计的元素做出评价，要求k个设计元素的满意度最大，输出元素的标号(降序输出)。

#include <iostream>
#include <algorithm>
using namespace std;
typedef struct 
{
    double data;
    int index;
}shirt;

int cmp(shirt a,shirt b)
{
 	if(a.data != b.data)
	 	return a.data > b.data;
	else
		return a.index < b.index;   
}
int cmp1(shirt a,shirt b)
{
    return a.index > b.index;
}
int main()
{
	shirt s[10000];
    int n,m,k;
    double temp;
    while(cin >> n >> m >> k)
    {
    	int i;
        for(i = 0;i < 10000;i++)
        {
            s[i].data = 0;
        }
        
        while(n--)
        {
            for(i = 0;i < m;i++)
            {
                cin >> temp;
                s[i].data += temp;
                s[i].index = i;
            }
        }
        sort(s,s+m,cmp);
        sort(s,s+k,cmp1);
        for(i = 0;i < k;i++)
        {
            cout << s[i].index+1;
            if(i < k-1)
            	cout << " "; /*不能使用ends,否则会WA，题意说明： There must be exactly one 
				             space between two adjacent indices,
							 and no extra space at the end of the line.*/
        }
        cout << endl;
    }
    return 0;
}

#include <iostream>
#include <cstdlib>
using namespace std;
typedef struct 
{
    double data;
    int index;
}shirt;
shirt s[1010];
int cmp(const void *a,const void *b)
{
    shirt *c = (shirt *)a;
    shirt *d = (shirt *)b;
    if(c->data == d->data) return c->index - d->index;
    else    return (d->data - c->data) > 0 ? 1:-1;//注意这里是double
}
int cmp1(const void *a,const void *b)
{
    shirt *c = (shirt *)a;
    shirt *d = (shirt *)b;
    return d->index - c->index;
}
int main(int argc, char *argv[])
{

    int n,m,k;
    double temp;
    while(cin >> n >> m >> k)
    {
        for(int i = 0;i <= m;i++)
        {
            s[i].data = 0;
        }
        
        while(n--)
        {
            for(int i = 0;i < m;i++)
            {
                cin >> temp;
                s[i].data += temp;
                s[i].index = i;
            }
        }
        qsort(s,m,sizeof(s[0]),cmp);
        qsort(s,k,sizeof(s[0]),cmp1);
        for(int i = 0;i < k-1;i++)
        {
            cout << s[i].index+1 << " ";
        }
        cout << s[k-1].index+1 << endl;
    }
    return 0;
} 

感想：这道题浪费好多时间，原因有两个：第一，不会使用结构题二级排序，第二，没有认真审题，ends不代表space，导致WA好多次；

以下代码是我最初的想法（提交就WA）：

思路：把n个人分别对m中设计元素的看法保存在二维数组map里面，遍历，是一列一列地遍历，找到最大值，然后再遍历一遍，把相同的最大值都改为-1，然后把最大值所在的列设为-1，最大值的列下表保存在数组put里面，排序后输出~

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

double map[1000][1000];
int n,m,k;
int cmp(const void *a,const void *b)
{
   return *(int *)b - *(int *)a;
}
void f()
{
    int put[1000];
    memset(put,0,sizeof(put));
    int index = 0;
    int s = k;
    while(s--)
    {
        double max = 0;
        int c;
        for(int i = 1;i <= m;i++)
        {
            for(int j = 1;j <= n;j++)
            {
                if(map[j][i] > max)
                {
                    max = map[j][i];
                    c = i;
                }
            }
        }
        for(int i = 1;i <= n;i++)
        {
            for(int j = 1;j <= m;j++)
            {
                if(max == map[i][j])
                    map[i][j] = -1;
            }
        }
        for(int i = 1;i <= n;i++)
            map[i][c] = -1;
       put[index] = c;
       index++;
    }
    qsort(put,k,sizeof(put[0]),cmp);
    for(int i = 0;i < k-1;i++)
    {
        cout << put[i] << " ";
    }
    cout << put[k-1] << endl;
}
int main()
{
    while(cin >> n >> m >> k)
    {
        memset(map,0,sizeof(map)); 
        for(int i = 1;i <= n;i++)
        {
            for(int j = 1;j <= m;j++)
            {
                cin >> map[i][j];
            }
        }
        f();
    }
    return 0;
}