【Java】リストからある条件のものを削除

Question

以下の自作クラスがあります。

public class Model {

	/** 時刻. */
	private LocalTime localTime;
	/** 身長 */
	private int height;
	/** 体重. */
	private int weight;

	public Model(LocalTime localTime, int height, int weight) {
		this.localTime = localTime;
		this.height = height;
		this.weight = weight;
	}
}

これをメインクラスでインスタンスを作成し、リストに入れて
時刻が重複しているものをある条件で削除したいのですがうまくできません。
ある条件は身長が低いほうを削除、身長が同じであれば体重が低いほうを削除。
身長も同じであれば、ランダムで1つ残す。
試しに以下のような感じで削除したのですが、リストの中身のオブジェクトが1つも残りませんでした。
どなたか教えて頂ける幸いです。

		List<Model> list = new ArrayList<>();

		Model a = new Model(LocalTime.of(0, 0), 60, 90);
		Model b = new Model(LocalTime.of(0, 0), 60, 90);
		Model c = new Model(LocalTime.of(12, 0), 65, 90);
		Model d = new Model(LocalTime.of(12, 0), 70, 80);
		Model e = new Model(LocalTime.of(15, 0), 80, 90);
		Model f = new Model(LocalTime.of(15, 0), 80, 90);

		list.add(a);
		list.add(b);
		list.add(c);
		list.add(d);
		list.add(e);
		list.add(f);

		for (int i = 0; i < list.size(); i++) {
			if (list.get(i).getLocalTime() == list.get(i + 1).getLocalTime()) {
				if (list.get(i).getTall() > list.get(i + 1).getTall()) {
					list.remove(i + 1);
				} else {
					list.remove(i);
				}
			}
		}

Comments

[jimbe]

Model には getLocalTime() 等のメソッドはありませんが。

Answer 1

1つの時刻に対して1つのオブジェクトだけ対応付けるという事なので、Mapの出番かなと思います。
「削除する」ではなく、「条件に合うものだけを集めたものを新たに作成する」という事になりますが、扱いやすくなります。
まずModelクラスにComparableを実装しましょう。比較がやりやすくなります。
hope_mucciさんがすでに実装例を示していますが、ここは直感的に従いやすくするために、時刻・身長・体重が大きいものを「大きい」と判定するようにしましょう。そうすると、同じ時刻内では「最も大きい」ものだけ残ればいいことになります。

public class Model implements Comparable {
    // 中略
    @Override
    public int compareTo(Model o) {
        int comp = this.localTime.compareTo(o.localTime);
        if (comp != 0) return comp;
        comp = Integer.compare(this.height, o.height);
        if (comp != 0) return comp;
        return Integer.compare(this.weight, o.weight);
    }

Map作成に移ります。LocalTimeをキーとしたマップにし、被ったら比較して大きいほうを残します。
「ランダムに残す」の部分は、この3要素が同じであれば結局同じものになるので無視します。

        Map<LocalDate, Model> map = new HashMap<>();
        for (Model m : list) {
            map.merge(m.getLocalDate(), m, (v1, v2) -> v1.compareTo(v2) < 0 ? v2 : v1);
        }
        List<Model> list2 = new ArrayList<>(map.values());

Comments

[taro_11111]

ありがとうございます。
comapareTo()をModel側で実装することによって、期待していた動作となりましたm(__)m

Answer 2

課題解決の方針

• Java 16のrecord型を使ってModelを定義する
• CollectorsのgroupingByによってLocalTimeが同じModelをグループ分けして、Listを作成する
• グループ分けされたListをソートまたはリダクションしてModelを１件求める。削除ではなく採用。
• Optional<Model>が発生しないようにする

Model

record型でかつComparableを実装する。Comparableは降順に並ぶよう定義する

import java.time.LocalTime;

public record Model(LocalTime localTime, int height, int weight) implements Comparable<Model> {
  @Override
  public int compareTo(Model o) {
    int lt = this.localTime.compareTo(o.localTime);
    if (lt != 0) return lt;
    int ht = Integer.compare(this.height, o.height);
    if (ht != 0) return -ht;
    return -Integer.compare(this.weight, o.weight);
  }
}

グループ分けすると、LocalTimeをキーにList<Model>を値とするマップができあがる。マップの各リストを降順にソートして先頭の1件を取る。

import java.time.LocalTime;
import java.util.List;
import java.util.stream.Collectors;

public class q356940 {
  
  public static void main(String[] args) {
    
    var list = List.of(
        new Model(LocalTime.of(0, 0), 60, 90),
        new Model(LocalTime.of(0, 0), 60, 90),
        new Model(LocalTime.of(12, 0), 65, 90),
        new Model(LocalTime.of(12, 0), 70, 80),
        new Model(LocalTime.of(15, 0), 80, 90),
        new Model(LocalTime.of(15, 0), 80, 91),
        new Model(LocalTime.of(15, 0), 80, 90)
    );
    
    var selected = list.stream()
                       .collect(Collectors.groupingBy(Model::localTime))
                       .values().stream()
                       .flatMap(l -> l.stream().sorted().limit(1))
                       .collect(Collectors.toList());
    
    selected.forEach(System.out::println);
    
  }

}

Collectors.minByを使うと簡単ですが、結果がOptional<Model>なので、それを避けるために、初期値を与えるリダクションを記述する必要があります。あるいは、OptionalのorElseを使う後処理を書くか。どちらも簡潔さは損なわれます。

Answer 3

既存のリストをあれこれいぢりまわすより、新しいリストをつくったほうがいいとおもいます。ということで、

リストからある条件のものを削除

という要件は満たしていませんが。

1. 時間ごとに分ける
2. 身長毎に分ける
3. 身長の一番でかいのをとる
4. 体重ごとに分ける
5. 体重の一番でかいのをとる
6. ランダムに一件選ぶ

# ま、自分向けにstreamのお勉強がてら書いてみました

import java.time.LocalTime;
import java.util.*;
import java.util.stream.Collectors;

public class xxHogege {

  public static void main(final String[] args) throws Exception {
    final List<Model> list = new ArrayList<>();

    Model a = new Model(LocalTime.of(0, 0), 60, 90);
    Model b = new Model(LocalTime.of(0, 0), 60, 90);
    Model c = new Model(LocalTime.of(12, 0), 65, 90);
    Model d = new Model(LocalTime.of(12, 0), 70, 80);
    Model e = new Model(LocalTime.of(15, 0), 80, 90);
    Model f = new Model(LocalTime.of(15, 0), 80, 90);

    list.add(a);
    list.add(b);
    list.add(c);
    list.add(d);
    list.add(e);
    list.add(f);
    list.add(new Model(LocalTime.of(15, 0), 66, 190));
    list.add(new Model(LocalTime.of(15, 0), 55, 290));

    System.out.println("------");
    System.out.println(list);

    final Map<LocalTime, List<Model>> localtimeMap =
        list.stream().collect(Collectors.groupingBy(x -> x.localTime));

    //    System.out.println("------ localtimeMap");
    //    System.out.println(localtimeMap);

    final List<Model> newList = localtimeMap.values().stream().map(modelList -> {
      //          System.out.println("------ modelList");
      //          System.out.println(modelList);

      final TreeMap<Integer, List<Model>> heightMap =
          new TreeMap(modelList.stream().collect(Collectors.groupingBy(x -> x.height)));
      final List<Model> maxHeightList = heightMap.get(heightMap.lastKey());
      //      System.out.println("------ maxHeightList " + heightMap.lastKey());
      //      System.out.println(maxHeightList);

      final TreeMap<Integer, List<Model>> weightMap =
          new TreeMap(maxHeightList.stream().collect(Collectors.groupingBy(x -> x.weight)));
      final List<Model> maxWeightList = weightMap.get(weightMap.lastKey());
      //      System.out.println("------ maxWeightList " + weightMap.lastKey());
      //      System.out.println(maxWeightList);

      return maxWeightList.get(new Random().nextInt(maxWeightList.size()));
    }).sorted(Comparator.comparing(x -> x.localTime)).collect(Collectors.toList());

    System.out.println("------ newList");
    System.out.println(newList);
  }

  public static class Model {
    /** 時刻. */
    final LocalTime localTime;
    /** 身長 */
    final int height;
    /** 体重. */
    final int weight;

    public Model(LocalTime localTime, int height, int weight) {
      this.localTime = localTime;
      this.height = height;
      this.weight = weight;
    }

    @Override
    public String toString() {
      return Map.of("localTime", localTime, "height", height, "weight", weight).toString();
    }

    @Override
    public boolean equals(Object obj) {
      return obj != null && this.getClass().equals(obj.getClass()) && this.localTime.equals(
          ((Model) obj).localTime) && this.height == ((Model) obj).height
          && this.weight == ((Model) obj).weight;
    }

    @Override
    public int hashCode() {
      return Objects.hash(localTime, height, weight);
    }
  }
}
//------
//[{localTime=00:00, height=60, weight=90}, {localTime=00:00, height=60, weight=90}, {localTime=12:00, height=65, weight=90}, {localTime=12:00, height=70, weight=80}, {localTime=15:00, height=80, weight=90}, {localTime=15:00, height=80, weight=90}, {localTime=15:00, height=66, weight=190}, {localTime=15:00, height=55, weight=290}]
//------ newList
//[{localTime=00:00, height=60, weight=90}, {localTime=12:00, height=70, weight=80}, {localTime=15:00, height=80, weight=90}]

Comments

[jimbe]

equals で toString を使っていることに関しまして、一応注意書きがあった方が良いように思いますが、如何でしょう。

[shiketa]

本筋に関係なさそうだからまいいか、と端折りましたった。手直ししておきました。

[jimbe]

お手数お掛けしてすいません。

[xebme]

これは力作ですね。

[shiketa]

ども f(^^;
\# 最近はKotlinあたまなので、Java/Streamの煩わしさがヤでヤで...
data classはうかんだのですが、recordはおもいうかばなかったです。それも含めて相当すっきりしますね。勉強になります。

Answer 4

データ処理にランダムを入れてくるのが新鮮ですので、そのままコード化してみます。

import java.time.LocalTime;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;

class Model {
  /** 時刻. */
  final LocalTime localTime;
  /** 身長 */
  final int height;
  /** 体重. */
  final int weight;

  public Model(LocalTime localTime, int height, int weight) {
    this.localTime = localTime;
    this.height = height;
    this.weight = weight;
  }

  @Override
  public String toString() {
    return new StringBuilder(super.toString())
        .append("[localtime=").append(localTime)
        .append(",height=").append(height)
        .append(",weight=").append(weight)
        .append("]").toString();
  }
}

public class Q356940 {
  public static void main(String[] args) {
    List<Model> list = new ArrayList<>();

    Model a = new Model(LocalTime.of(0, 0), 60, 90);
    Model b = new Model(LocalTime.of(0, 0), 60, 90);
    Model c = new Model(LocalTime.of(12, 0), 65, 90);
    Model d = new Model(LocalTime.of(12, 0), 70, 80);
    Model e = new Model(LocalTime.of(15, 0), 80, 90);
    Model f = new Model(LocalTime.of(15, 0), 80, 90);

    list.add(a);
    list.add(b);
    list.add(c);
    list.add(d);
    list.add(e);
    list.add(f);
    for(Model m : list) System.out.println(m);

    Random random = new Random();
    for (int i = 0; i < list.size(); i++) {
      for (int j = i+1; j < list.size(); j++) {
        //時刻比較(どちらかが消える)
        if (list.get(i).localTime == list.get(j).localTime) {
          //身長比較(小さいほうが消える)
          if (list.get(i).height > list.get(j).height) {
            list.remove(j--); continue;
          } else if (list.get(i).height < list.get(j).height) {
            list.remove(i--); break;
          }
          //体重比較(軽いほうが消える)
          if (list.get(i).weight > list.get(j).weight) {
            list.remove(j--); continue;
          } else if (list.get(i).weight < list.get(j).weight) {
            list.remove(i--); break;
          }
          //どっちでもいい(ランダムに消える)
          if (random.nextInt(2) == 0) {
            list.remove(j--); continue;
          } else {
            list.remove(i--); break;
          }
        }
      }
    }

    System.out.println("----");
    for(Model m : list) System.out.println(m);
  }
}

Model@73a28541[localtime=00:00,height=60,weight=90]
Model@470e2030[localtime=00:00,height=60,weight=90]
Model@3fb4f649[localtime=12:00,height=65,weight=90]
Model@33833882[localtime=12:00,height=70,weight=80]
Model@200a570f[localtime=15:00,height=80,weight=90]
Model@16b3fc9e[localtime=15:00,height=80,weight=90]
----
Model@73a28541[localtime=00:00,height=60,weight=90]
Model@33833882[localtime=12:00,height=70,weight=80]
Model@200a570f[localtime=15:00,height=80,weight=90]

Answer 5

このような場合、Model側にcompareToメソッドを実装してデータ同士の大小関係を比較するのが定石になります。
専用のインターフェースがあります。
https://docs.oracle.com/javase/jp/8/docs/api/java/lang/Comparable.html

    /**
     * 引数のModelと比較する。時刻、身長、体重の順で大小を比較する
     * 正の値が返れば、this側のオブジェクトの方が大きい。
     * 負の値が返れば、other側のオブジェクトの方が大きい。
     */
    public int compareTo(Model other){
      // まず時刻の比較
      int comp;
      comp = this.localTime.compareTo(other.getLocalTime());
      if(comp != 0)
        return comp;
      // 時刻が同じ(compが0)の場合、身長で比較
      comp = this.height - other.getHeight();
      if(comp != 0)
        return comp;
      // 身長も同じ(compが0)の場合、体重で比較
      return this.weight - other.getWeight();
    }

Main側ではまずリスト中の2つの要素について時刻を比較し、異なっていたらそのまま残す。
同じであれば、上記作成したcompareToメソッドで身長、体重を比較し、小さい方を削除するようにすれば良いでしょう。
（これは元のリストが時刻でソートされていることが前提になります）

        for (int i = 0; i < list.size()-1; i++) {
          Model m1 = list.get(i);
          Model m2 = list.get(i+1);
          // 2つの時刻が違っていたらそのまま残す
          if(!m1.getLocalTime().equals(m2.getLocalTime())){
            continue;
          }else{
            // 時刻が同じだったら、compareToメソッドで身長、体重の大小を比較
            int comp = m1.compareTo(m2);
            // compが正ならm1が大きいのでm2を削除、、compが負ならm2が大きいのでm1を削除
            if(comp > 0){
              list.remove(m2);
              i--;
            }else if(comp < 0){
              list.remove(m1);
              i--;
            }else{
              // 同じだったらランダムで削除とのことだが、実装が面倒なのでm1を削除する
              list.remove(m1);
              i--;
            }
          }
        }

paiza.ioでのデモ

Comments

[taro_11111]

すみません、comparetoが既に時刻を昇順に並び替える事で使用していました。。。

Answer 6

先頭から順に探した場合、あるインデックスのものを削除すると、それ以降のものは順番が狂ってしまいます

こういう場合の鉄則は、後尾から先頭へ向かって探していくこと、です。
後ろからだと、削除してもそれ以降の順番は狂いません