1025. 反转链表 (25)--做题记录

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⁵)、以及正整数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

#include <iostream>
#include <map>
#include <stack>
using namespace std;

void appendZero(int data) {
    if (data < 10) {
      printf("0000%d",data);
    }else if (data < 100) {
      printf("000%d",data);
    }else if (data < 1000) {
      printf("00%d",data);
    }else if (data < 10000) {
      printf("0%d",data);
    }else {
      printf("%d",data);
    }
}

int main() {

  map<int, int>valueMap, nextMap;

  int begin, N, K;
  cin>>begin>>N>>K;

  //cout<<begin<<" "<<N<<" "<<K;
  int i = 0;
  for (i = 0; i < N; i++) {
    int address, data, next;
    cin>>address>>data>>next;
    valueMap.insert(pair<int, int>(address, data));
    nextMap.insert(pair<int, int>(address, next));
  }

  //cout<<"-----------------------------------"<<endl;

  stack<int> s1;

  int count = 0, index = begin, waitToPrintNext = 0;
  while(true) {
    
    // 计数
    count++;
      
    // 压栈
    s1.push(index);

    if (count == K) {
      for (i = 0; i < K; i++){

        if (i == 0 && waitToPrintNext == 1) {
          appendZero(s1.top());
          printf("\n");
          waitToPrintNext = 0;
        }

        appendZero(s1.top());
        printf(" %d ", valueMap[s1.top()]);
        s1.pop();

        if (i != K - 1) {
          appendZero(s1.top());
          printf("\n");
        }
      }
      count = 0;
      waitToPrintNext = 1;
    }

    // index 后移
    index = nextMap[index];

    if (index == -1) {
      if (s1.size() == 0) {
        printf("-1\n");
      }
      break;
    }
  }

  // 余下结尾

  stack<int> s2;

  while (s1.size() > 0){
    s2.push(s1.top());
    s1.pop();
  }

  i=0;
  while (s2.size() > 0){
    if (i==0&&waitToPrintNext==1) {
      appendZero(s2.top());
      printf("\n");
      i++;
    }

    appendZero(s2.top());
    printf(" %d ", valueMap[s2.top()]);
    if (s2.size() != 1) {
      appendZero(nextMap[s2.top()]);
    }else {
      printf("-1");
    }
    printf("\n");
    s2.pop();
  }


  return 0;
}