PAT 1110 区块反转 25分

1110 区块反转 - PAT (Basic Level) Practice （中文） (pintia.cn)

解析：1110 区块反转 – PAT乙级真题 – 柳婼 の blog (liuchuo.net)

坑点：

生成链表之后要更新链表的节点个数，因为输入的节点可能是无效节点，我们需要筛选一下。

#include<iostream>
#include<vector>
using namespace std;
const int N=1e5+10;
struct Node
{
  int data ;int next;  
}node[N];

vector<int>L,block[N],ans;

int address,tail,c,cnt;
int main()
{
int head1,n,k;
cin>>head1>>n>>k;

    //首先读入链表
    for(int i=0;i<n;i++)
    {
        cin>>address;
        cin>>node[address].data>>node[address].next;
    }

    //生成链表L
    tail=head1;
    while(tail!=-1)
    {
        L.push_back(tail);
        tail=node[tail].next;
    }
    //重新统计链表节点个数
    n=L.size();

  //遍历链表L，每k个块加入到链表E中
    for(int i=0;i<n;i++)
    {
        block[c].push_back(L[i]);  //注意这里，一个block[c]就加入了一个k块
        cnt++;
        if(cnt==k&&i!=n-1)
        {
            cnt=0;
            c++;  //block[c++]=k++;准备加入下一个k块
        }
    }

    //从后往前，每k个块加入到答案链表中
    for(int i=c;i>=0;i--)
    {
        for(auto& it:block[i])
        {
            ans.push_back(it);
        }
    }


    //输出
    for(int i=1;i<n;i++)
    {
        printf("%05d %d %05d\n",ans[i-1],node[ans[i-1]].data,ans[i]);
    }
    printf("%05d %d -1",ans.back(),node[ans.back()].data);
    
    
    
    return 0;
}