PAT乙級B1025 反轉連結串列 (25)
阿新 • • 發佈:2019-02-09
給定一個常數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;
}