PAT甲级P1074题解

#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
int result[N];
int datas[N];
int List[N];
int nexts[N];
int main(void)
{
    int first,n,k,temp;
    cin>>first>>n>>k;
    for(int i = 0;i<n;i++)
    {
        cin>>temp;
        cin>>datas[temp]>>nexts[temp];
    }
    int sum = 0;
    while(first !=-1)
    {
        List[sum++] = first;
        first = nexts[first];
    }
    for(int i = 0;i<sum;i++)
    {
        result[i] = List[i];
    }
    for(int i = 0;i<(sum-sum%k);i++)
    {
        result[i] = List[i / k * k + k - 1 - i%k];
    }
    for(int i = 0;i<sum-1;i++)
    {
        printf("%05d %d %05d\n",result[i],datas[result[i]],result[i+1]);
    }
    printf("%05d %d -1",result[sum-1],datas[result[sum-1]]);
    return 0;
}

通过sum来统计节点数目,并且通过 i = i/k*k+k-1-i%k来将节点的地址反转