C++で連結リストのノードをペアごとに交換する方法
ここでは、連結リスト(リンクリスト)が与えられたとき、隣り合う2つのノードをペアごとに入れ替え、その結果の先頭ノードを返す問題を扱います。重要な制約として、ノードが持つ値(val)を変更することはできず、ノード自体のつながり(ポインタ)だけを操作して入れ替えを行う必要があります。
例えば、リストが [1,2,3,4] の場合、処理後のリストは [2,1,4,3] となります。奇数個の要素がある場合は、最後の1つのノードはそのまま残ります。
アルゴリズムの手順
この問題は、以下の手順で解くことができます。
- head が存在しない(NULL の)場合は、そのまま head を返す
- first := head、second := head の次のノード とする。また、値 -1 を持つダミーノード dummy を新しく作成する
- dummy->next := first とし、prev := dummy と初期化する
- second が NULL でない限り、以下を繰り返す
- temp := second の次のノード を保存しておく
- first->next := second->next とする
- second->next := first とする(first と second を入れ替え)
- prev->next := second とする(前のペアと新しいペアをつなぐ)
- prev := first と更新する
- temp が NULL でなければ、first := temp、second := temp->next として次のペアへ進む。そうでなければループを抜ける
- 最後に dummy->next を返す(これが新しい先頭ノードになる)
C++での実装例
それでは、実際の実装を見て理解を深めましょう。
#include <bits/stdc++.h>
using namespace std;
class ListNode{
public:
int val;
ListNode *next;
ListNode(int data){
val = data;
next = NULL;
}
};
ListNode *make_list(vector<int> v){
ListNode *head = new ListNode(v[0]);
for(int i = 1; i<v.size(); i++){
ListNode *ptr = head;
while(ptr->next != NULL){
ptr = ptr->next;
}
ptr->next = new ListNode(v[i]);
}
return head;
}
void print_list(ListNode *head){
ListNode *ptr = head;
cout << "[";
while(ptr->next){
cout << ptr->val << ", ";
ptr = ptr->next;
}
cout << "]" << endl;
}
class Solution {
public:
ListNode* swapPairs(ListNode* head) {
if(!head)return head;
ListNode* first= head;
ListNode* second = head->next;
ListNode* dummy = new ListNode(-1);
dummy->next = first;
ListNode* prev = dummy;
while(second){
ListNode* temp = second->next;
first->next = second->next;
second->next = first;
prev->next = second;
prev = first;
if(temp){
first = temp;
second = temp ->next;
}
else break;
}
return dummy->next;
}
};
main(){
Solution ob;
vector<int> v = {1,2,3,4,5,6,7,8};
ListNode *head = make_list(v);
print_list(ob.swapPairs(head));
}入力
[1,2,3,4,5,6,7,8]
出力
[2,1,4,3,6,5,8,7]
ポイントの解説
この実装の鍵となるのはダミーノード(dummy)の活用です。先頭ノード自体が入れ替わるため、戻り値となる新しい先頭を一意に管理できるようになります。また、各ペアの処理では次の順序でポインタを張り替えています。
- second の次のノードを temp として退避させる
- first の next を second の次に向けて、first をペアの後ろに移動させる
- second の next を first に向けて、second をペアの先頭にする
- 前のペアの末尾(prev)の next を second につなぎ直して、リスト全体を連結する
この手法により、ノードの値を一切書き換えることなく、ポインタの付け替えだけで隣接ノードのペア交換を実現できます。計算量はリストの長さを n とすると O(n)、追加のメモリ使用量は O(1) です。
-
C++で二分木の葉ノードをペアごとに入れ替える方法
二分木が与えられたとき、その葉ノードをペアごとに入れ替えることが課題です。例えば、次のような入出力が考えられます。 入力: 出力: 本記事では、隣接する2つの葉ノードを指す2つのポインタを管理しながら、それらの値を順番に入れ替えていく方法を解説します。 解法のアプローチ このアプローチでは、木を走査して葉ノードを見つけると同時に、カウンタで現在の葉ノードの個数を追跡します。ポイントとなるのは次のロジックです。 カウンタが奇数のとき:まだペアの片方しか見つかっていないため、1つ目のポインタをそのノードに向ける。 カウンタが偶数のとき:ペアが揃ったことになるので、2つのノードのデータを入れ
-
C++で木の全ノードに中間順後続ノード(Inorder Successor)を設定する方法
この問題では、nextポインタを持つ木構造が与えられます。私たちのタスクは、このnextポインタに各ノードの中間順後続ノード(inorder successor)を設定することです。まず、使用するノード構造体は以下のように定義されています。struct node { int value; struct node* left; struct node* right; struct node* next; }すべてのnextポインタは初期状態でNULLに設定されており、これを各ノードの中間順後続ノードを指すように書き換える必要があります。基本概念のおさらい中間順走査