マップで集合演算(和集合・積集合・差集合)をしよう

このレッスンでは、Goのマップを使って和集合・積集合・差集合を求める方法を学び、集合演算の基本を理解することを目標にします。「Go言語マップ集合演算」で調べている方向けです。

Goには専用のSet型はありませんが、map[int]boolを使うと集合のように扱えます。それぞれのキーの有無を調べることで、「和集合」「積集合」「差集合」を計算できます。

サンプルコードでは、{1, 2, 3, 4}と{3, 4, 5, 6}という2つのマップに対して、それぞれの要素の有無を確認しながら和集合・積集合・差集合の結果を組み立て、sort.Intsで見やすい順序に並び替えてから表示しています。

初心者がよくつまずくのが、マップの中身の順序が保証されない点です。表示順序をそろえたい場合は、キーだけを取り出したスライスを別途sortする必要がある点に注意しましょう。

「両方のグループに共通するユーザー」を見つけたいときなど、実務のデータ処理でも役立つ考え方で、積集合や差集合の考え方はデータベースのSQLクエリにおける結合操作の理解にもつながります。

📖 お手本コード

package main

import (
	"fmt"
	"sort"
)

func main() {
	a := map[int]bool{1: true, 2: true, 3: true, 4: true}
	b := map[int]bool{3: true, 4: true, 5: true, 6: true}

	union := map[int]bool{}
	for k := range a {
		union[k] = true
	}
	for k := range b {
		union[k] = true
	}

	intersection := map[int]bool{}
	for k := range a {
		if b[k] {
			intersection[k] = true
		}
	}

	difference := map[int]bool{}
	for k := range a {
		if !b[k] {
			difference[k] = true
		}
	}

	printSorted := func(m map[int]bool) {
		keys := []int{}
		for k := range m {
			keys = append(keys, k)
		}
		sort.Ints(keys)
		fmt.Println(keys)
	}

	printSorted(union)
	printSorted(intersection)
	printSorted(difference)
}

✍️ あなたのコード

コードを書いたら「実行する」を押してください

🧪 このサイトではGoを直接コンパイル・実行できないため、お手本コードと同じ内容を入力できたかをその場で判定します(採点はブラウザ内だけで行われ、どこにも送信されません)。

🎯

ここまでできればゴール達成!
2つのグループの共通点や違いを調べる処理が作れるようになります。