浙江大学PAT甲级A1074题解(C++)(1)

#include <iostream>
#include <map>
#include <cstdio>

using namespace std;
int main() {
    int first,n,k;
    cin>>first>>n>>k;
//    int temp;
    int temp, data[100005], next[100005], list[100005], result[100005];
    int sum =0;
    for(int i=0;i<n;i++){
        cin>>temp;
        cin>>data[temp]>>next[temp];
    }

    while (first!=-1){
        //list按顺序保存当前结点地址
        list[sum++]=first;
        first = next[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)*2+1)*k-1-i];
           
    }
    for (int i = 0; i < sum - 1; i++) {
        printf("%05d %d %05d\n", result[i], data[result[i]], result[i + 1]);
    }
    printf("%05d %d -1", result[sum - 1], data[result[sum - 1]]);
    return 0;



}

下面详细说明此算法精髓
result[i] = list[((i/k)*2+1)*k-1-i];
设两个数组下标i,x。则i+x=k-1;i+x=3k-1;i+x=5k-1…
则i+x = nk -1
显然！
n=2(i/k)+1（i/k=0,n=1;i/k=1,n=3）
x=((i/k)*2+1)*k-1-i
这里的x就是result[x]，这里的值要被list对应位置的值替换