1. 程式人生 > >PAT乙級B1025 反轉連結串列 (25)

PAT乙級B1025 反轉連結串列 (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^5^)、以及正整數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

思路:

1.由於地址只有5位,不妨開個陣列,第一次遍歷直接以地址作為下標,儲存連結串列結點。然後使用雙端佇列 每k個節點逆序輸出一次。具體用法參考:雙端佇列deque用法介紹

2.注意k為0或1時連結串列按原順序輸出。另外輸出第一個節點時注意特判,與後面格式不同。

參考程式碼:

#include<cstdio>
#include<deque>
using namespace std;
const int maxn=100010;
struct node{
	int addr,data,next;	//分別表示連結串列結點當前地址,資料,下一結點地址
}List[maxn],temp;
int main()
{
	int n,start,k,p,cnt=0;
	deque<node> q;
	scanf("%d%d%d",&start,&n,&k);
	for(int i=0;i<n;i++)	//這個迴圈儲存所有結點
	{
		scanf("%d%d%d",&temp.addr,&temp.data,&temp.next);
		List[temp.addr]=temp;
	}
	p=start;
	bool flag=false;	//用來標記第一個輸出的結點
	while(p!=-1)	//這個迴圈用來逆轉並輸出連結串列結點
	{
		cnt++;
		q.push_back(List[p]);
		p=List[p].next;
		if(k>1&&cnt%k==0){	//k小於2直接原順序輸出
			while(!q.empty())
			{
				node t=q.back();
				q.pop_back();
				if(!flag){
					printf("%05d %d",t.addr,t.data);
					flag=true;
				}else{
					printf(" %05d\n%05d %d",t.addr,t.addr,t.data);
				}
			}
		}
		
	}
	while(!q.empty())		//最後不足k個節點時或k小於2時直接原順序輸出
	{
		node t=q.front();
		q.pop_front();
		if(!flag){
			printf("%05d %d",t.addr,t.data);
			flag=true;
		}else{
			printf(" %05d\n%05d %d",t.addr,t.addr,t.data);
		}
	}
	printf(" -1\n");
	return 0;
}

優化:

1.直接使用reverse函式進行反轉操作(標頭檔案是algorithm)。

2.連結串列中儲存的是地址。

參考程式碼:

#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=100010;
int main()
{
	int n,st,k,sum=0,addr,data[maxn],next[maxn];
	vector<int> List;
	scanf("%d%d%d",&st,&n,&k);
	for(int i=0;i<n;i++)	
	{
		scanf("%d",&addr);
		scanf("%d %d",&data[addr],&next[addr]);
	}
	while(st!=-1){
		List.push_back(st);
		sum++;
		st=next[st];
		if(k>1&&sum%k==0)
			reverse(List.begin()+sum-k,List.begin()+sum);
	}
	for(int i=0;i<sum-1;i++)
	{
		printf("%05d %d %05d\n",List[i],data[List[i]],List[i+1]);
	}
        printf("%05d %d -1\n", List[sum - 1], data[List[sum - 1]]);
	return 0;
}