java 文字の一致がわからない。

Question

###前提・実現したいこと
赤 1
青 2
黄色 3
上記のように数えた回数を表示したい。

###発生している問題・エラーメッセージ

赤 1 
青 2 
黄色 3 
青 1 
黄色 2 
黄色 1

###該当のソースコード

import java.util.*;


public class Main {
    public static void main(String[] args) {
        

        Scanner sc = new Scanner(System.in);
        String a = "赤 青 黄色 青 黄色 黄色";
        String[] b = a.split("[ ]");
        int n[] = new int[b.length];
        
        
        for(int i=0;i<b.length;i++) {
            for(int j=i;j<b.length;j++) {
                if(b[i].equals(b[j])) {
                    n[i] = n[i] + 1;
                }
            }
        }
        
        for(int i=0;i<b.length;i++) {
            System.out.println(b[i] + " " + n[i]);
        }
    }
}

###試したこと
最後のfor文がいけないことはわかっているのですが、どうしたら「実現したいこと」のような結果になりますか。

###補足情報(言語/FW/ツール等のバージョンなど)
より詳細な情報

Answer 1

例えばi=1の時のループでj=3の時にifを満たし配列の数値をインクリメントします。この時点で本来ならb[3]にある「青」は用済みなのですが、そのまま残っているためi=3のループで再度カウントされてしまいます。
###単純に考えると
先述の通り、一度探したものはもう用済みなので、配列から消してしまうのが簡単でしょうか。
この場合、一致検索する際に「消えているかどうか」の判定も必要になります。

public class Main {
    public static void main(String[] args) {


        Scanner sc = new Scanner(System.in);
        String a = "赤 青 黄色 青 黄色 黄色";
        String[] b = a.split("[ ]");
        int n[] = new int[b.length];


        for(int i = 0; i < b.length; i++) {
            if (b[i] == null) continue;
            n[i] = 1;
            for(int j = i + 1; j < b.length; j++) {
                if(b[i].equals(b[j])) {
                    n[i] = n[i] + 1;
                    b[j] = null;
                }
            }
        }

        for(int i = 0; i < b.length; i++) {
            if (b[i] == null) continue;
            System.out.println(b[i] + " " + n[i]);
        }
    }
}

###Mapを使って単純化
Mapを使えば、同じもののカウントを簡単にできます。

public class Main {
    public static void main(String[] args) {


        Scanner sc = new Scanner(System.in);
        String a = "赤 青 黄色 青 黄色 黄色";
        String[] b = a.split("[ ]");
        Map<String, Integer> map = new HashMap<>();

        for(int i = 0; i < b.length; i++) {
            map.merge(b[i], 1, Integer::sum);
        }

        map.forEach((k, v) -> System.out.println(k + " " + v));
    }
}

###Streamを使ってさらに短く

public class Main {
    public static void main(String[] args) {

        String a = "赤 青 黄色 青 黄色 黄色";
        String[] b = a.split("[ ]");
        Arrays.stream(b).collect(Collectors.groupingBy(s -> s, Collectors.counting()))
                        .forEach((k, v) -> System.out.println(k + " " + v));
    }
}

Answer 2

最後のforがいけないんではなくて、

        for(int i=0;i<b.length;i++) {
            for(int j=i;j<b.length;j++) {
                if(b[i].equals(b[j])) {
                    n[i] = n[i] + 1;
                }
            }
        }

ここの処理がちょっと意味合いが違ってます。
この処理だと自身とそれ以後の一致する数のカウントになります。
これでも出力時にうまいことすればできますが、その場合はこの文字のカウントはすでに表示したから飛ばすとかの処理が必要になります。

が、まあこういうのはさくっと思いつく処理としては、Hashって言う物があるのでこれを利用してカウントするのがいいかと思います。

import java.util.*;

public class Main {
	public static void main(String[] args) {
		Map<String, Integer> map = new HashMap<String, Integer>();
		for(String val : "赤 青 黄色 青 黄色 黄色".split(" ")) {
			if(map.containsKey(val) == false) {
			}
			map.put(val, map.get(val) + 1);
		}
		
		for (String key : map.keySet()) {
			System.out.println(key + " " + map.get(key));
		}
	}
}

※あえて一部処理を飛ばしてエラーにしてますので、そこは勉強がてら調べてみてください。

Comments

[num79kill]

最後のログはentrySetでやろうよ。