题目
拍集体照时队形很重要,这里对给定的 N 个人 K 排的队形设计排队规则如下:
每排人数为 N/K(向下取整),多出来的人全部站在最后一排;
后排所有人的个子都不比前排任何人矮;
每排中最高者站中间(中间位置为 m/2+1,其中 m 为该排人数,除法向下取整);
每排其他人以中间人为轴,按身高非增序,先右后左交替入队站在中间人的两侧(例如5人身高为190、188、186、175、170,则队形为175、188、190、186、170。这里假设你面对拍照者,所以你的左边是中间人的右边);
若多人身高相同,则按名字的字典序升序排列。这里保证无重名。
现给定一组拍照人,请编写程序输出他们的队形。
输入格式:
每个输入包含 1 个测试用例。每个测试用例第 1 行给出两个正整数 N(≤10^4,总人数)和 K(≤10,总排数)。随后 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
思路
将所有人由高到矮排序,身高相同的按姓名排,再逐行放置即可。
对于每一行,先找到中间位置放第1个,再从中间按左、右的次序依次放置剩下的。用变量d控制距离中间的距离,每次变换符号,如果变号后小于0则再增加距离。
int mid = len / 2;
names[mid] = p[count++].name;
int d = -1;
for (int j=1; j<len; j++){
names[mid+d] = p[count++].name;
d = -d;
if (d<0){
d--;
}
}
注意行数大于人数的情况,此时所有人集中到1排,排完1排即可结束。
代码
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
struct people{
string name;
int height;
};
bool cmp(people p1, people p2){
if (p1.height==p2.height){
return p1.name < p2.name;
}
else {
return p1.height > p2.height;
}
}
int main() {
int n, k;
cin >> n >> k;
vector<people> p(n);
for (int i=0; i<n; i++) {
cin >> p[i].name >> p[i].height;
}
sort(p.begin(), p.end(), cmp);
int count = 0;
for (int i=0; i<k; i++){
int len = n / k;
if (i==0){
len += n % k;
}
if (len==0){
break;
}
vector<string> names(len);
int mid = len / 2;
names[mid] = p[count++].name;
int d = -1;
for (int j=1; j<len; j++){
names[mid+d] = p[count++].name;
d = -d;
if (d<0){
d--;
}
}
cout << names[0];
for (int j=1; j<len; j++){
cout << " " << names[j];
}
cout << endl;
}
return 0;
}