[paiza]シリをもげ！（みんなでしりとり (paizaランク B 相当)

大人になれない僕らの

無敵フォルゴレと強力わかもとビクトリームがめっちゃ好き。
キャッチマイハート、ベリーメロン（ﾍﾞﾘｰﾒﾛﾝ
ぶるぁあぁあぁあ～っ！！

チェック処理とかは端折ってる

/** ここから定型文 */
ここで記載したクラスについては、
こっちの記事に書いてある、標準入力をList化したりしてるMainクラスから呼び出す予定だし・なんか継承したりしてるYO!!

ちな、Mainを実行して標準入力から入力するのクソ面倒くさいので、
Java編については問題の入力例をパラメータにしたテストクラスとかも作って公開するYO!!

開発・実行環境はこんな感じ

• VSCode
• Java 17
• jUnit 5.9
• maven

pom.xmlはこんな感じ

pom.xml

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>jp.co.asil</groupId>
    <artifactId>paiza202408</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.source>17</maven.compiler.source>
        <maven.compiler.target>17</maven.compiler.target>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.junit.jupiter</groupId>
            <artifactId>junit-jupiter-engine</artifactId>
            <version>5.9.1</version>
            <scope>test</scope>
        </dependency>
    </dependencies>
</project>

/** ここまで定型文 */

実装クラス

WordChain.java

package jp.co.asil.paiza202408;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class WordChain extends Question {

  public WordChain(List<String> list) {
    super(list);
  }

  /**
   * あなたは友達たちと N 人でしりとりを行うことにしました。
   * 1 人目、 2 人目、...、 N 人目、 1 人目、2 人目、... という順序で発言をします。
   * 
   * ここで、それぞれの人は、次に挙げる 4 つのしりとりのルールを守って発言をする必要があります。
   * 
   * 1. 発言は、単語リストにある K 個の単語のうちのいずれかの単語でなければならない。
   * 2. 最初の人以外の発言の頭文字は、直前の人の発言の最後の文字と一緒でなければならない。
   * 3. 今までに発言された単語を発言してはならない。
   * 4. z で終わる単語を発言してはならない。
   * 
   * ここで、発言の途中で上のルールを破った場合、ルールを破った人はしりとりから外れます。
   * そして、その人を抜いて引き続きしりとりを続けていきます。このとき、後続の人は、ルール 2 を守る必要はありません。
   * 
   * N 人がしりとりを行ったログが M 行分与えられます。
   * このとき、M 回の発言が終わった後、しりとりから脱落せずに残っている人のリストを表示するプログラムを書いてください。
   */
  @Override
  public List<String> answer() {

    int[] cond = List.of(list.get(0).split(" ")).stream().mapToInt(Integer::parseInt).toArray();
    // 選択可能単語リスト
    List<UseWord> words = list.subList(1, cond[1] + 1).stream().map(UseWord::new).toList();
    boolean[] isStart = { true };
    String preWord = "";
    // プレイヤー
    List<Integer> players = IntStream.range(1, cond[0] + 1).boxed().collect(Collectors.toList());
    int palyerIndex = 0;
    // ログ処理
    List<String> logList = list.subList(cond[1] + 1, list.size());
    for (String log : logList) {
      UseWord w = words.stream().filter(word -> word.word.equals(log)).findFirst().orElse(null);
      boolean isErr = true;
      if (w != null) {
        isErr = w.isSelected || log.endsWith("z")
            || (!isStart[0] && !log.startsWith(preWord.substring(preWord.length())));
        w.isSelected = true;
      }
      if (isErr) {
        preWord = "";
        isStart[0] = true;
        List<Integer> newPlayer = new ArrayList<>();
        for (int i = 0; i < players.size(); i++) {
          if (i != palyerIndex) {
            newPlayer.add(players.get(i));
          }
        }
        players = newPlayer;
      } else {
        preWord = log;
        isStart[0] = false;
        palyerIndex++;
        if (palyerIndex == players.size()) {
          palyerIndex = 0;
        }
      }
    }
    // 残ってる人数と残ってる人の番号を設定
    List<String> result = new ArrayList<>();
    result.add(String.valueOf(players.size()));
    result.addAll(players.stream().map(String::valueOf).toList());
    return result;
  }

  class UseWord {
    String word;
    boolean isSelected;

    public UseWord(String word) {
      this.word = word;
      this.isSelected = false;
    }
  }
}

実装クラス

WordChainTest.java

package jp.co.asil.paiza202408;

import static org.junit.jupiter.api.Assertions.assertArrayEquals;

import java.util.List;

import org.junit.jupiter.api.Test;

public class WordChainTest {
  @Test
  void testAnswer1() {
    WordChain testClass = new WordChain(List.of("3 6 7",
        "a",
        "aloha",
        "app",
        "az",
        "paiza",
        "warp",
        "app",
        "paiza",
        "a",
        "aloha",
        "az",
        "warp",
        "paiza"));
    assertArrayEquals(testClass.answer().toArray(), new String[] { "1",
        "3" });
  }

  @Test
  void testAnswer2() {
    WordChain testClass = new WordChain(List.of("4 4 4",
        "abacus",
        "banana",
        "candy",
        "yankee",
        "banana",
        "candies",
        "candies",
        "yankee"));
    assertArrayEquals(testClass.answer().toArray(), new String[] { "2",
        "1",
        "4" });
  }
}

なんでこんなに

JavaScriptだと楽に書けるのに、
なんでJavaだとこんなに行数が増えてしまったのか。。。
Javaはこの業界は言った当初からイジってるし、最近はラムダで書くのを勧めてはいるけど、拡張ForやラムダのforEachでindexが使えない時点でゴミ言語って思ってしまうわ。。。