二重ループ：基本編

ループ・二重ループ

hogeちゃん

2022.07.05

二重ループメニュー【指定された範囲・行数の数字の出力・二次元配列での要素の検索・最大の要素・行ごとの和・積の最大ほか】

問題【STEP: 1】指定された数字までの出力(空白区切り)

Task

1 〜 N の整数を 1 から順に半角スペース区切りで 1 行で出力してね。

【STEP: 1】コード

reader.on('close', () => {
   const N = Number(lines[0]);
   const A = [];
   for (let i = 1; i <= N; i++) {
      A.push(i);  // A に i(1〜N)を追加
   }
   console.log(A.join(' '));  // 連結して出力
});

hogeちゃんの画像

for文でカウンタ変数 i の値を配列 A に保存後 join(' ') で値を連結して出力しました。

問題【STEP: 2】指定された数字までの出力(改行区切り)

Task

1 〜 N の整数を 1 から順に改行区切りで N 行出力してね。

【STEP: 2】コード

reader.on('close', () => {
   const N = Number(lines[0]);
   for (let i = 1; i <= N; i++) {
      console.log(i);  // i(1〜N)を出力
   }
});

hogeちゃんの画像

カウンタ変数 i の値を for文でループしながら出力しました。

問題【STEP: 3】指定された行数の数字の出力

Task

1 から 5 までの数字を半角スペース区切りしたものを N 行出力してね。

【STEP: 3】コード

reader.on('close', () => {
   const N = Number(lines[0]);
   for (let i = 1; i <= N; i++) {
      const A = [];
      for (let j = 1; j <= 5; j++) {
         A.push(j);
      }
      console.log(A.join(' '));
   }
});

hogeちゃんの画像

1〜5 の値を持つ配列 A を N 回出力しています。

問題【STEP: 4】指定された範囲・行数の数字の出力

Task

1 から N までの数字を半角スペース区切りしたものを K 行出力してね。

【STEP: 4】コード

reader.on('close', () => {
   const [N, K] = lines[0].split(' ').map(Number);
   for (let i = 1; i <= K; i++) {
      const A = [];
      for (let j = 1; j <= N; j++) {
         A.push(j);
      }
      console.log(A.join(' '));
   }
});

hogeちゃんの画像

1〜N の値を持つ配列 A を K 行出力。

問題【STEP: 5】二次元配列の出力

Task

N 行 K 列の二次元配列 A が与えられるのでそのまま出力してね。

【STEP: 5】コード

reader.on('close', () => {
   const [N, K] = lines[0].split(' ').map(Number);
   const A = [];
   for (let i = 1; i <= N; i++) {
      A.push(lines[i].split(' ').map(Number));
      console.log(A[i - 1].join(' '));
   }
});

hogeちゃんの画像

2次元配列化しましたが出力は join(' ') を使ったので2重ループにはならなかった…。(^_^;

問題【STEP: 6】二次元配列での要素の検索

Task

N 行 K 列の二次元配列 A の要素内で 1 になっている要素の行と列を出力してね。

【STEP: 6】コード

reader.on('close', () => {
   const [N, K] = lines[0].split(' ').map(Number);
   for (let i = 1; i <= N; i++) {
      const A = lines[i].split(' ').map(Number);
      for (let j = 0; j < K; j++) {
         if (A[j] === 1) {
            console.log(i, j + 1);
            break;
         }
      }
   }
});

hogeちゃんの画像

最初のループで lines[i]を配列に変換して A に保存。2つ目のループで A の各値にアクセスし 1 であれば i と j + 1 を表示しています。

問題【STEP: 7】二次元配列の最大の要素

Task

N 行 K 列の二次元配列 A の要素のうち最大の要素の値を出力してね。

【STEP: 7】コード

reader.on('close', () => {
   const [N, K] = lines[0].split(' ').map(Number);
   let max = 0;
   for (let i = 1; i <= N; i++) {
      const A = lines[i].split(' ').map(Number);
      for (let j = 0; j < K; j++) {
         if (A[j] > max) {
            max = A[j];
         }
      }
   }
   console.log(max);
});

hogeちゃんの画像

最初のループで lines[i]を配列に変換して A に保存。2つ目のループで A の各値にアクセスして変数 max より大きければ max の値を更新しました。

問題【STEP: 8】二次元配列の行ごとの和

Task

N 行 K 列の二次元配列 A の行ごとの和を出力してね。

【STEP: 8】コード

reader.on('close', () => {
   const [N, K] = lines[0].split(' ').map(Number);
   for (let i = 1; i <= N; i++) {
      const A = lines[i].split(' ').map(Number);
      console.log(A.reduce((a, c) => a + c));
   }
});

hogeちゃんの画像

便利な reduce() 使ってます。

問題【STEP:9】さまざまな長さの配列の和

Task

配列 A のサイズ K と A の要素が1行 ✕ N 行与えられるので要素の和を求めて出力してね。

【STEP:9】コード

reader.on('close', () => {
   const N = Number(lines[0]);
   for (let i = 1; i <= N; i++) {
      const A = lines[i].split(' ').map(Number);
      A.shift();
      console.log(A.reduce((a, c) => a + c));
   }
});

hogeちゃんの画像

問題の趣旨と合わないような気がしますが入力の最初の値を削除して要素の和を出力しています。。

問題【STEP:10】階段の出力

Task

1 から i (1〜N)までの数字を半角スペース区切りで N 行出力してね。

【STEP:10】コード

reader.on('close', () => {
   const N = Number(lines[0]);
   const A = [];
   for (let i = 1; i <= N; i++) {     
      A.push(i);     
      console.log(A.join(' '));
   }
});

hogeちゃんの画像

A の値を増やしながら出力しています。

問題【FINAL】積の最大

Task

(配列 A の 1 つの要素) × (配列 B の 1 つの要素) の最大値を求めてね。

【FINAL】コード

reader.on('close', () => {
   const [N, K] = lines[0].split(' ').map(Number);
   const A = lines[1].split(' ').map(Number);
   const B = lines[2].split(' ').map(Number);
   let max = -100 * 100;  // max の初期値は入力条件の 最小 ✕ 最大
   for (let i = 0; i < N; i++) {
      for (let j = 0; j < K; j++) {
         const n = A[i] * B[j];
         if (n > max) {
            max = n;
         }
      }
   }
   console.log(max);   
});

hogeちゃんの画像

出力値の初期値は入力条件の範囲内での積の最小値に設定しました。2重ループで全ての積の結果を比較して max より大きければ max を上書きしています。

コメント