pat乙级1055
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
#include<iostream>//利用大顶堆，定义好排序规则
#include<queue>
#include<string>
#include<vector>
#include<cmath>
using namespace std;
struct stu
{
string name;
int top;
friend bool operator<(const stu& s1, const stu& s2)
{
if (s1.top != s2.top)
return s1.top < s2.top;//大的top在前（堆的结构体排序定义和sort相反）
else
return s1.name > s2.name;//top相等时，字典序小的在前
}
};
int main()
{
int n, k;
stu s;
priority_queue<stu, vector<stu> >q;//大顶堆
while (cin >> n >> k)
{
while (!q.empty())q.pop();
for (int i = 0; i < n; i++)
{
cin >> s.name >> s.top;
q.push(s);
}
int pou = floor(n / k);
int lastpou = n - (k-1) * pou;//先安排最后一排，即输出的第1排
int lastmid = lastpou / 2 + 1;
vector<stu>p(lastpou + 1);//0位不放数据
int flag = -1;//先左后右的输入方法j,j-1,j+2,j-3,j+4,以此类推
for (int i = 1, j = lastmid; j > 0 && j <= lastpou; i++)
{
p[j] = q.top();
q.pop();
j = j + i * flag;
flag = -flag;
}
vector<vector<stu> >a(k - 1, vector<stu>(pou+1));
int mid = pou / 2 + 1;
for (int i = 0; i < k - 1; i++)
{
flag = -1;
for (int u = 1, j = mid; j > 0 && j <= pou; u++)
{
a[i][j] = q.top();
q.pop();
j = j + u * flag;
flag = -flag;
}
}
for (int i = 1; i <= lastpou; i++)
{
if (i != lastpou)
cout << p[i].name<<" ";
else
cout<< p[i].name <<endl;
}
for (int i = 0; i < k - 1; i++)
{
for (int j = 1; j <= pou; j++)
{
if (j != pou)
cout << a[i][j].name << " ";
else
cout << a[i][j].name << endl;
}
}
}
return 0;
} 代码学习笔记 文章被收录于专栏
学习笔记，pat,牛客
腾讯一面1122人在聊
投递美团等公司10个岗位 >