PAT Basic 1055 集体照 (25 分)

拍集体照时队形很重要,这里对给定的 N 个人 K 排的队形设计排队规则如下:

  • 每排人数为 /(向下取整),多出来的人全部站在最后一排;

  • 后排所有人的个子都不比前排任何人矮;

  • 每排中最高者站中间(中间位置为 /,其中 m 为该排人数,除法向下取整);

  • 每排其他人以中间人为轴,按身高非增序,先右后左交替入队站在中间人的两侧(例如5人身高为190、188、186、175、170,则队形为175、188、190、186、170。这里假设你面对拍照者,所以你的左边是中间人的右边);

  • 若多人身高相同,则按名字的字典序升序排列。这里保证无重名。

现给定一组拍照人,请编写程序输出他们的队形。

输入格式:

每个输入包含 1 个测试用例。每个测试用例第 1 行给出两个正整数 N(≤,总人数)和 K(≤,总排数)。随后 N 行,每行给出一个人的名字(不包含空格、长度不超过 8 个英文字母)和身高([30, 300] 区间内的整数)。

输出格式:

输出拍照的队形。即K排人名,其间以空格分隔,行末不得有多余空格。注意:假设你面对拍照者,后排的人输出在上方,前排输出在下方。

输入样例:

10 3
Tom 188
Mike 170
Eva 168
Tim 160
Joe 190
Ann 168
Bob 175
Nick 186
Amy 160
John 159

输出样例:

Bob Tom Joe Nick
Ann Mike Eva
Tim Amy John


算法思路:首先进行排序,然后进行算出几行几列,可以求余数,用vector进行补在第一行。
接着,进行一次遍历,使用双端队列,进行左插右插交替进行即可。
#include <iostream>
#include <vector>
#include <deque>
#include <algorithm>
using namespace std;
struct stu{
    string name;
    int len;
};
bool cmp(stu s1,stu s2){
    if(s1.len!=s2.len) return s1.len>s2.len;
    return s1.name<s2.name;
}
int main(){
    int M,N;
    cin>>M>>N;
    stu s[M];
    for(int i=0;i<M;i++)
        cin>>s[i].name>>s[i].len;
    sort(s,s+M,cmp);
    int row=M/N;int more=M%N;
    vector<vector<stu>> v;bool start=true;
    int k=0;
    if(M<N){
        vector<stu> tmp;
        for(int i=0;i<M;i++) tmp.push_back(s[i]);
        v.push_back(tmp);
    }else{
        while(k<M){
            vector<stu> tmp;
            if(start){
                start=false;
                for(int i=0;i<row+more;i++,k++) tmp.push_back(s[k]);
                v.push_back(tmp);
            }else{
                for(int i=0;i<row;i++,k++) tmp.push_back(s[k]);
                v.push_back(tmp);
            }
        }
    }
    vector<deque<stu>> res;
    for(int i=0;i<v.size();i++){
        deque<stu> tmp_que;
        tmp_que.push_back(v[i][0]);bool dir=0;//0是左插
        for(int j=1;j<v[i].size();j++){
            if(!dir) {
                tmp_que.push_front(v[i][j]);
                dir=!dir;
            }else{
                tmp_que.push_back(v[i][j]);
                dir=!dir;
            }
        }
        res.push_back(tmp_que);
    }

    for(int i=0;i<res.size();i++){
        for(int j=0;j<res[i].size();j++)
            if(j!=res[i].size()-1) cout<<res[i][j].name<<" ";
            else cout<<res[i][j].name<<endl;
    }
    system("pause");
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/littlepage/p/11701678.html