paiza 問題集ソートと検索 query js

クエリメニュー【要素の追加・要素の検索・要素の削除・連想配列・ソートと検索】

Task

配列 A が与えられるので A[K] の後ろに Q を挿入した後改行区切りで出力してね。

【STEP: 1】コード

reader.on('close', () => {
   const [N, K, Q] = lines[0].split(' ').map(Number);
   const A = [];
   for (let i = 1; i <= N; i++) {
      A.push(lines[i]);
   }
   A.splice(K, 0, Q);  // A の K 番目の要素を 0 個削除して Q を追加
   A.forEach(a => {
      console.log(a);
   });
});

A.splice(追加・削除を行う位置, 削除する要素数, 追加する値)。

top

問題【STEP: 2】指定要素の検索

Task

数列 A に K が含まれていれば″YES″・そうでなければ″NO″を出力してね。

【STEP: 2】コード

reader.on('close', () => {
   const [N, K] = lines[0].split(' ').map(Number);
   const A = [];
   for (let i = 1; i <= N; i++) {
      A.push(Number(lines[i]));
   }
   console.log(A.includes(K) ? 'YES' : 'NO');  // A に K は含まれる? trueなら 'YES' false なら 'NO'
});

A.includes(K) で配列 A に K が含まれているかどうかの真偽値が取得できました。

top

問題【STEP: 3】指定要素の検索 (query)

Task

数列 A と Q 個の整数(K_1〜K_Q)が与えられるので A に K_i が含まれているかを判定してね。

【STEP: 3】コード

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

ループしながら A.includes(K)で配列 A に K が含まれているかどうかの真偽値を取得。

top

問題【STEP: 4】先頭の要素の削除

Task

数列 A の先頭の要素を削除した後の A を出力してね。

【STEP: 4】コード

reader.on('close', () => {
   const N = Number(lines[0]);
   const A = [];
   for (let i = 1; i <= N; i++) {
      A.push(Number(lines[i]));
   }
   A.shift();  // 先頭の要素を削除
   A.forEach(a => {
      console.log(a);
   });
});

A.shift()で配列 A の先頭の要素を削除しています。

top

問題【STEP: 5】先頭の要素の削除(query)

Task

数列 A に対し K 回の入力に応じてた処理をしてね。(pop=先頭を削除・show=出力)

【STEP: 5】コード

reader.on('close', () => {
   const [N, K] = lines[0].split(' ').map(Number);
   const A = [];
   for (let i = 1; i <= N; i++) {
      A.push(Number(lines[i]));
   }
   for (let i = N + 1; i <= N + K; i++) {
      const S = lines[i];
      switch (S) {
         case 'pop':  // S が "pop" なら…
            A.shift();  // 先頭の要素を削除
            break;
         case 'show':  // S が "show" なら…
            A.forEach(a => {
               console.log(a);  // A の各要素を出力
            });
            break;
      }
   }
});

switch文で処理を振り分けて ″pop″であれば A.shift() を実行。″show″であれば A の各要素を出力しています。

top

問題【STEP: 6】連想配列

Task

生徒の出席番号と識別 ID を覚えて各番号に対応する生徒の識別 ID を出力してね。

【STEP: 6】コード

reader.on('close', () => {
  const [N, K] = lines[0].split(' ').map(Number);
  const ID = new Map();  // マップ ID を宣言
  for (let i = 1; i <= N; i++) {
    const [num, id] = lines[i].split(' ');
    ID.set(num, id);  // ID に 要素 {num(キー), id(値)} を保存
  }
  for (let i = N + 1; i <= K + N; i++) {
      const key = lines[i];
    console.log(ID.get(key));  // ID の key とペアになっている値を出力
  }
});

マップ ID を宣言し ID.set(num, id) で num = キー, id = 値をペアで保存。 ID.get(key) で入力された key とペアになった値を取得。

top

問題【STEP: 7】連想配列(query)

Task

生徒の出席番号と ID を覚えてイベントに応じて処理(覚える・忘れる・出力)してね。

【STEP: 7】コード

reader.on('close', () => {
   const [N, K] = lines[0].split(' ').map(Number);
   const ID = new Map();  // マップ ID を宣言
   for (let i = 1; i <= N; i++) {
      const [num, id] = lines[i].split(' ');
      ID.set(num, id);  // ID に 要素 {num(キー), id(値)} を保存
   }
   for (let i = N + 1; i <= K + N; i++) {
      const S = lines[i].split(' ');
      switch (S[0]) {
         case 'join':  // S[0] が "join" なら…
            ID.set(S[1], S[2]);  // ID に 要素 {S[1](キー), S[2](値)} を追加
            break;
         case 'leave':  // S[0] が "leave" なら…
            ID.delete(S[1]);  // ID から S[1](キー) とペアになっている要素を を削除
            break;
         case 'call':  // S[0] が "call" なら…
            console.log(ID.get(S[1]));  // ID の S[1](キー) とペアになっている値を 出力
            break;
      }
   }
});

switch文で処理内容を振り分けて ID に要素 ({S[1](キー), S・[2](値)} )を追加したり・要素の削除・出力を実行しています。

top

問題【STEP: 8】ソートと検索

Task

転校生が入ってきた後クラス全員で背の順で並んだ時 paiza 君は何番目？

【STEP: 8】コード

reader.on('close', () => {
   const [N, X, P] = lines[0].split(' ').map(Number);
   const A = [];
   for (let i = 1; i <= N; i++) {
      A.push(Number(lines[i]));  // A に クラスの身長を保存
   }
   A.push(X, P);  // A に X, P を 追加
   A.sort((s, b) => s - b);
   console.log(A.indexOf(P) + 1);  // A に X, P を 追加
});

A.indexOf(P) で配列 A に含まれる P の index が取得できました。

top

問題【FINAL】ソートと検索 (query)

Task

paiza 君のクラス…イベントに応じて処理 (加入・背の順に並んで出力) してね。

【FINAL】コード

reader.on('close', () => {
  const [N, K, P] = lines[0].split(' ').map(Number);
  const A = [];
  for (let i = 1; i <= N; i++) {
    A.push(Number(lines[i]));  // A に クラスの身長を保存
  }
  A.push(P);  // A に P を追加
  for (let i = N + 1; i <= K + N; i++) {
    const S = lines[i].split(' ');
    switch (S[0]) {
      case 'join':  // S[0] が "join" なら…
         A.push(S[1]);  // A に S[1] を追加
         break;
      case 'sorting':  // S[0] が "sorting" なら…
         A.sort((s, b) => s - b);  // A を昇順で並替え
         console.log(A.indexOf(P) + 1);  // A に含まれる P の 添字 + 1 を出力
      break;
    }
  }
});

イベントが 2 種類発生するので switch文で処理を振り分け。

top