L2-022 重排連結串列

YuKiCheng發表於2024-03-18

這道題真的煩,輸出想半天。
反正就是要區分奇偶,才能知道那個結點最後要列印出-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;
}

相關文章