题意:给一个链表的所有节点,链表头,现在要求把负数全排在前面,0-k中的数排在中间,其它的排在最后,要求相对位置不变。
思路:map映射一下地址和节点,剩下的用vector模拟一下就好。
代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <vector>
using namespace std;
struct Data {
int add, v, nxt;
Data(int _add, int _v, int _nxt) : add(_add), v(_v), nxt(_nxt) {}
Data() {}
};
map<int, Data> mp;
int fir, n, k;
int tfir, tend, tv;
vector<Data> v;
vector<Data> ans;
int main() {
// freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);
scanf("%d %d %d", &fir, &n, &k);
for (int i = 0; i < n; i++) {
scanf("%d %d %d", &tfir, &tv, &tend);
mp[tfir] = Data(tfir, tv, tend);
}
int p = fir;
while (p != -1) {
v.push_back(mp[p]);
p = mp[p].nxt;
}
for (int i = 0; i < v.size(); i++) {
if (v[i].v < 0) ans.push_back(v[i]);
}
for (int i = 0; i < v.size(); i++) {
if (0 <= v[i].v && v[i].v <= k) ans.push_back(v[i]);
}
for (int i = 0; i < v.size(); i++) {
if (k < v[i].v) ans.push_back(v[i]);
}
for (int i = 0; i < ans.size(); i++) {
if (i == ans.size()-1) printf("%05d %d -1\n", ans[i].add, ans[i].v);
else {
printf("%05d %d %05d\n", ans[i].add, ans[i].v, ans[i+1].add);
}
}
return 0;
}