PAT 1055. 集体照
拍集体照时队形很重要,这里对给定的N个人K排的队形设计排队规则如下:
每排人数为N/K(向下取整),多出来的人全部站在最后一排;
后排所有人的个子都不比前排任何人矮;
每排中最高者站中间(中间位置为m/2+1,其中m为该排人数,除法向下取整);
每排其他人以中间人为轴,按身高非增序,先右后左交替入队站在中间人的两侧(例如5人身高为190、188、186、175、170,则队形为175、188、190、186、170。这里假设你面对拍照者,所以你的左边是中间人的右边);
若多人身高相同,则按名字的字典序升序排列。这里保证无重名。
现给定一组拍照人,请编写程序输出他们的队形。
输入格式:
每个输入包含1个测试用例。每个测试用例第1行给出两个正整数N(<=10000,总人数)和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
分析
直接按题意排序,然后一排一排的输出。
看下栗子:
如果要排的数从大到小排好了,且他们的指针(在这理解为下标)为 0,1,2,3,4,5(这一排有6个人)
则输出对应的下标为5,3,1,0,2,4
如果要排的数从大到小排好了,且他们的指针(在这理解为下标)为 0,1,2,3,4,5,6(这一排有7个人)
则输出对应的下标为5,3,1,0,2,4,6
不难看出每排的情况若以0即中间最高的分左右两边来看,左边全是奇数,且递减,右边全是偶数,递增;
在利用迭代器,可将每排的情况都变成这样;
代码如下
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct stud{
string name;
int height;
};
bool compare(stud s1,stud s2){
return s1.height!=s2.height?s1.height>s2.height:s1.name<s2.name;
}
int main(){
int N,K;
cin>>N>>K;
vector<stud> vi(N);
for(int i=0;i<N;i++)
cin>>vi[i].name>>vi[i].height;
sort(vi.begin(),vi.end(),compare);
vector<int> row(K,N/K);
row[K-1]+=N%K; // 最后一排要加上可能多出来的人
int first,r=K-1;
for(auto b=vi.begin();b<vi.end();b+=row[r--]){
if(row[k]%2==0) first=row[r]-1; // first来确定每排第一个要输出的人
else first=row[r]-2;
for(;first>0;first-=2)
cout<<(b+first)->name<<" ";
cout<<b->name; // first=0时一定是最高的拿个
for(first=2;first<=row[k]-1;first+=2)
cout<<" "<<(b+first)->name;
cout<<endl;
}
return 0;
}