PAT乙级1055 集体照//双指针运用

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

• 每排人数为 N/K（向下取整），多出来的人全部站在最后一排；
• 后排所有人的个子都不比前排任何人矮；
• 每排中最高者站中间（中间位置为 m/2+1，其中 m 为该排人数，除法向下取整）；
• 每排其他人以中间人为轴，按身高非增序，先右后左交替入队站在中间人的两侧（例如5人身高为190、188、186、175、170，则队形为175、188、190、186、170。这里假设你面对拍照者，所以你的左边是中间人的右边）；
• 若多人身高相同，则按名字的字典序升序排列。这里保证无重名。
  现给定一组拍照人，请编写程序输出他们的队形。

输入格式：
每个输入包含 1 个测试用例。每个测试用例第 1 行给出两个正整数 N（≤10⁴，总人数）和 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

思路：我的方法较为繁琐（可是一次AC真的好激动，但是看到柳神45行解决我…）这题关键就是如何将学生按题目要求的顺序排好，我使用双指针解决。
假设一个数组vec，其中呈递减排列（因为我是降序排好再传过来的），index指向vec头。再新建一个ans数组，mid指向ans的中间，left指向mid左边一格，right指向mid右边一格。刚开始将index指向的元素复制到left中，left左移，index前移；再将index的元素复制到right中，right右移，index前移…以此类推。
函数实现：

void change(vector<stu> &vec)
{//注意：传递过来的vec内元素呈降序排列
	int len = vec.size();
	vector<stu> ans(len);
	int mid = ans.size() / 2;
	//实际是ans.size() / 2 - 1 + 1，-1是因为下标从0开始，+1是因为取mid/2+1
	int index = 0, left = mid - 1, right = mid + 1, flag = 1;
	ans[mid] = vec[index++]; //mid保存vec第一个元素（即最大的元素），然后index自增
	while (index < vec.size())
	{
		switch(flag)
		{
		case 1: //先填入left指向的地方
			{
				ans[left--] = vec[index++];
				break;
			}
		case -1: //后填入right指向的地方
			{
				ans[right++] = vec[index++];
				break;
			}
		}
		flag *= -1; //flag取相反数
	}
	for (int i = 0; i < len; i++)
		vec[i] = ans[i];
}

总代码：

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
using namespace std;
struct stu
{
	string name;
	int height;
};
bool cmp(stu a, stu b)
{
	if (a.height != b.height)
		return a.height > b.height;
	else
		return a.name < b.name;
}
void change(vector<stu> &);
int main()
{
	int n, k;
	cin >> n >> k;
	int eachp = n / k; //eachp表示每排的人数
	vector<stu> vec(n); //vec存储所有学生
	vector<vector<stu> > row(k); //row存储k排数组
	for (int i = 0; i < n; i++)
		cin >> vec[i].name >> vec[i].height;
	sort(vec.begin(), vec.end(), cmp); //排序
	
	int index = 0;
	for (int i = 0; i < k; i++)
	{
		if (i == 0 && eachp * k < n) //eachp * k小于n说明有多出来的人
		{
			int last = eachp + (n - eachp * k); //last表示最后一排的总人数
			for (int j = 0; j < last; j++)
				row[i].push_back(vec[index++]); //row[0]存储最高的一排人
			continue;
		}
		for (int j = 0; j < eachp; j++)
			row[i].push_back(vec[index++]);
	}
	
	for (int i = 0; i < k; i++)
	{
		change(row[i]); //先排序
		for (int j = 0; j < row[i].size(); j++)
		{
			cout << row[i][j].name;
			if (j != row[i].size() - 1)
				cout << ' ';
		}
		cout << endl;
	}
	return 0;
}
void change(vector<stu> &vec)
{
	int len = vec.size();
	vector<stu> ans(len);
	int mid = ans.size() / 2;
	int index = 0, left = mid - 1, right = mid + 1, flag = 1;
	ans[mid] = vec[index++];
	while (index < vec.size())
	{
		switch(flag)
		{
		case 1:
			{
				ans[left--] = vec[index++];
				break;
			}
		case -1:
			{
				ans[right++] = vec[index++];
				break;
			}
		}
		flag *= -1;
	}
	for (int i = 0; i < len; i++)
		vec[i] = ans[i];
}