1. 程式人生 > >乙級 PAT 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(<= 105)、以及正整數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<stdio.h>
#include<stdlib.h>

struct link {
	int head;
	int num;
	int tail;
}links[100000], link2[100000];



int main() {
	int head, tail, num, n, i;
	scanf("%d%d%d", &head, &n, &num);

	for (i = 0; i < n; i++) {
		scanf("%d%d%d", &links[i].head, &links[i].num, &links[i].tail);
	}
	int j = 0;
	while (head != -1) {
		for (i = 0; i < n; i++) {
			if (links[i].head == head) {
				link2[j] = links[i];
				j++;
				head = links[i].tail;
				break;
			}
		}
	}
	n = j;
	if (n%num == 0) {
		for (i = 1; i <= n / num; i++) {
			for (j = 1; j<num; j++)
				printf("%05d %d %05d\n", link2[num*i - j].head, link2[num*i - j].num, link2[num*i - 1 - j].head);
			if (i != n / num)
				printf("%05d %d %05d\n", link2[num*i - num].head, link2[num*i - num].num, link2[num*i + num - 1].head);
			else
				printf("%05d %d -1\n", link2[num*i - num].head, link2[num*i - num].num);
		}

	}
	else {
		for (i = 1; i <= n / num; i++) {
			for (j = 1; j<num; j++)
				printf("%05d %d %05d\n", link2[num*i - j].head, link2[num*i - j].num, link2[num*i - 1 - j].head);
			if (i != n / num)
				printf("%05d %d %05d\n", link2[num*i - num].head, link2[num*i - num].num, link2[num*i + num - 1].head);
			else
				printf("%05d %d %05d\n", link2[num*i - num].head, link2[num*i - num].num, link2[n / num*num].head);
		}

		for (i = n / num*num; i < n - 1; i++) {
			printf("%05d %d %05d\n", link2[i].head, link2[i].num, link2[i].tail);
		}

		printf("%05d %d -1\n", link2[n - 1].head, link2[n - 1].num);

	}

	return 0;
}