PAT 1109 Group Photo (25分)

题目很长，意思很简单。但是最近好像读题非常非常不仔细，老是犯很傻的错误。误以为K是除最后一行每行应该站的人数，然后只过三个点。后面仔细一看才知道写错了。

代码

#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>
#include <algorithm>
#include <vector>
#define MAX 10100
using namespace std;
int N,K;
struct Person
{
    string name;
    int height;
    bool operator < (Person x)
    {
        if(height==x.height)
            return name<x.name;
        return height>x.height;
    }
}person[MAX];
vector<string> vec[MAX];
int main()
{
    scanf("%d%d",&N,&K);
    for(int i=0;i<N;i++)
    {
        cin>>person[i].name>>person[i].height;
    }
    sort(person,person+N);
    int last = N%K + N/K;
    int ind = 1;
    if(last)
    {
        vec[0].resize(last);
        int center = last/2;
        vec[0][center] = person[0].name;
        int le = center-1;
        int ri = center+1;
        while(ind<last)
        {
            vec[0][le--] = person[ind++].name;
            if(ind>=last)
                break;
            vec[0][ri++] = person[ind++].name;
        }
    }
    int rows = K;
    int nums = N/K;
    for(int i=1;i<rows;i++)
    {
        vec[i].resize(nums);
        int center = nums/2;
        vec[i][center] = person[ind++].name;
        int le = center-1;
        int ri = center+1;
        while(ind<last+i*nums)
        {
            vec[i][le--] = person[ind++].name;
            if(ind>=last+i*nums)
                break;
            vec[i][ri++] = person[ind++].name;
        }
    }
    for(int i=0;i<K;i++)
    {
        for(int j=0;j<vec[i].size();j++)
        {
            if(j!=0)
                printf(" ");
            cout<<vec[i][j];
        }
        printf("\n");
    }
    return 0;
}