1025 反转链表(25) - PAT乙级

1025 反转链表(25)

给定一个常数 K 以及一个单链表 L,请编写程序将 L 中每 K 个结点反转。例如:给定 L 为 1→2→3→4→5→6,K 为 3,则输出应该为 3→2→1→6→5→4;如果 K 为 4,则输出应该为 4→3→2→1→5→6,即最后不到 K 个元素不反转。

输入格式:

每个输入包含 1 个测试用例。每个测试用例第 1 行给出第 1 个结点的地址、结点总个数正整数 N (≤10^5)、以及正整数 K (≤N),即要求反转的子链结点的个数。结点的地址是 5 位非负整数,NULL 地址用 −1 表示。

接下来有 N 行,每行格式为:
Address Data Next
其中 Address 是结点地址,Data 是该结点保存的整数数据,Next 是下一结点的地址。

输出格式:

对每个测试用例,顺序输出反转后的链表,其上每个结点占一行,格式与输入相同。

输入样例:

00100 6 4
00000 4 99999
00100 1 12309
68237 6 -1
33218 3 00000
99999 5 68237
12309 2 33218

输出样例:

00000 4 33218
33218 3 12309
12309 2 00100
00100 1 99999
99999 5 68237
68237 6 -1

Solution:

map+node存放结点, 再按链表顺序添加到vector中, 对vector每个局部进行reverse, 之后打印每个结点信息, 打印之前需要更新next

通过,新链表的总长N和给定结点间隔K, 能求出”链表”如果哪些位置反转.

for (int i = 0; i < v.size() / K; i++) {
reverse(v.begin() + i*k, v.begin() + (i+1)*k);
}

reverse函数的用法:

vector<int> v = {1, 2, 3, 4, 5, 6};
reverse(v.begin(), v.begin() + 3);
for (int i = 0; i < v.size(); i++) {
printf("%d ", v[i]);
}
printf("\n");
// 3 2 1 4 5 6

坑点:
1, 如果测试给出多余的结点的话, 新链表的总长不为N了; 我之前认为我用vector存放了node, v.size()是绝对安全的, 是, 明枪易躲, 暗箭难防, 对于这题最重要的reverse会出错; 如果不改好这个, case6通不过.

2, unordered_map<char*, Node> mp; 将node存入之后, 用字符数组作为键值是无法查找到的。如果要用字符串作为key值的话, map定义时用string不用char*, 暂时就这么理解

#include <cstdio>
#include <cstring>
#include <string>
#include <unordered_map>
#include <vector>
#include <algorithm>
using namespace std;


struct Node {
char addr[6];
int data;
char next[6];
};

int main() {
char head[6]; int N, K;
scanf("%s %d %d", head, &N, &K);
// unordered_map<char*, Node> mp; // error, why?
unordered_map<string, Node> mp;
for (int i = 0; i < N; i++) {
Node node;
scanf("%s %d %s", node.addr, &node.data, node.next);
mp[node.addr] = node;
}

vector<Node> v;
char pNode[6]; strcpy(pNode, head);
while (strcmp(pNode, "-1") != 0) {
v.push_back(mp[pNode]);
strcpy(pNode, mp[pNode].next); // debug01
}
for (int i = 0; i < v.size()/K; i++) { // case6
reverse(v.begin()+i*K, v.begin()+i*K+K); // 哨兵元素, i从0开始
}
int flag = 0;
for (int i = 0; i < v.size(); i++) {
if (i != v.size()-1) {
strcpy(v[i].next, v[i+1].addr);
} else {
strcpy(v[i].next, "-1");
}
if (flag) printf("\n");
printf("%s %d %s", v[i].addr, v[i].data, v[i].next);
flag = 1;
}

return 0;
}
/*
case6:
00100 4 2
00100 1 12309
12309 2 23333
77777 7 22222
23333 3 -1
*/