L2-022 重排链表

这道题真的烦,输出想半天。
反正就是要区分奇偶,才能知道那个结点最后要打印出-1.
我看网上遇到的都是测试点3的问题,不过我有问题的是测试点1,前三个出问题就是节点数奇偶的问题。

#include <bits/stdc++.h>
using namespace std; 
map<int, pair<int, int>> mp;
int main() {
	int s, n;
	cin >> s >> n;//起始和节点数量
	for (int i = 0; i < n; i++) {
		int a, b, c;
		cin >> a >> b >> c;
		mp[a] = { b,c };//b是值 c是下一个地址的值
	}
	vector<int> vec;//vec放的是从前往后依次的地址
	while (s != -1) {
		vec.push_back(s);
		s = mp[s].second;
	}
	if (vec.size() == 1) {
		printf("%05d %d -1\n", vec[0], mp[vec[0]].first);
		return 0;
	}
	int i = 0, j = vec.size() - 1;
	if (vec.size() % 2 == 0) {//如果是偶数
		for (int i = 0; i < vec.size()/ 2; i++) {
			int s1 = vec[i];
			int s2 = vec[vec.size() - 1 - i];
			printf("%05d %d %05d\n", s2, mp[s2].first, s1);
			if (i != vec.size()/ 2 - 1) {
				printf("%05d %d %05d\n", s1, mp[s1].first, vec[vec.size() - 1 - i-1]);
			}
			else {
				printf("%05d %d -1\n", s1, mp[s1].first);
			}
		}
	}
	else {
		for (int i = 0; i < vec.size()/2; i++) {
			int s1 = vec[i];
			int s2 = vec[vec.size() - 1 - i];
			printf("%05d %d %05d\n", s2, mp[s2].first, s1);
			printf("%05d %d %05d\n", s1, mp[s1].first, vec[vec.size() - 1 - i-1]);
		}
		int s2 = vec[vec.size() / 2];
		printf("%05d %d -1\n", s2, mp[s2].first);
	}
	return 0;
}
posted @ 2024-03-18 19:25  YuKiCheng  阅读(13)  评论(0编辑  收藏  举报