JavaScriptで文字列の二次元配列をソートして対角要素を見つける方法

本記事では、文字列の配列を扱うJavaScriptのアルゴリズム問題を解説します。「配列をアルファベット順にソートした後、対角線上の文字を抽出する」というシンプルながら応用範囲の広いテクニックを、サンプルコードとともにわかりやすく紹介します。

問題

n個の文字列を要素として持つ配列を受け取るJavaScript関数を作成します。ここで、配列内の各文字列はすべてちょうどn文字で構成されているものとします。つまり、この配列はn×nの正方行列として扱うことができます。

関数には以下の2つの処理が求められます。

• まず、配列をアルファベット順（辞書順）にソートすること
• 次に、ソート後の配列を行列とみなし、左上隅から始まる主対角線上にある文字を抽出して連結した文字列を返すこと

コード例

以下が実際のコードです。

const arr = [
   'star',
   'abcd',
   'calm',
   'need'
];
const sortPickDiagonal = () => {
   const copy = arr.slice();
   copy.sort();
   let res = '';
   for(let i = 0; i < copy.length; i++){
      for(let j = 0; j < copy[i].length; j++){
         if(i === j){
            res = res + copy[i][j];
         };
      };
   };
   return res;
};
console.log(sortPickDiagonal(arr));

出力結果

aaer

処理の流れの解説

このコードでは、まずslice()メソッドで元の配列のコピーを作成しています。こうすることで、元の配列を変更せずに済むのがポイントです。続いてsort()メソッドにより、配列がアルファベット順に並べ替えられます。

ソート後の配列は ['abcd', 'calm', 'need', 'star'] となります。二重ループの中で、行インデックスiと列インデックスjが一致している場合、すなわち対角要素である場合のみ、その文字を結果の文字列に追加していきます。

今回の例では、以下の4文字が対角要素として抽出されます。

• 0行0列目：「abcd」の「a」
• 1行1列目：「calm」の「a」
• 2行2列目：「need」の「e」
• 3行3列目：「star」の「r」

これらを連結すると「aaer」となり、コンソールに出力されます。なお、条件判定を利用すれば反対側の対角線（副対角線）の要素を取得することも可能なので、ぜひ応用してみてください。