From 60715cad1af02963536c5c4b75673d370ced6792 Mon Sep 17 00:00:00 2001
From: Antoine Martin <antoine97.martin@gmail.com>
Date: Sun, 6 Dec 2020 12:48:22 +0100
Subject: [PATCH] 2020: day06: part 2

---
 aoc2020/src/day06.rs | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/aoc2020/src/day06.rs b/aoc2020/src/day06.rs
index 7e224ad..fc58c5b 100644
--- a/aoc2020/src/day06.rs
+++ b/aoc2020/src/day06.rs
@@ -7,6 +7,7 @@ pub fn run() -> aoc::Result<String> {
     let mut res = String::with_capacity(128);
 
     writeln!(res, "part 1: {}", part1(INPUT)?)?;
+    writeln!(res, "part 2: {}", part2(INPUT)?)?;
 
     Ok(res)
 }
@@ -49,6 +50,24 @@ fn part1(input: &str) -> aoc::Result<usize> {
         .sum())
 }
 
+fn part2(input: &str) -> aoc::Result<usize> {
+    let groups = get_groups(input)?;
+
+    Ok(groups
+        .iter()
+        .map(|group| {
+            group
+                .answers
+                .iter()
+                .skip(1)
+                .fold(group.answers[0].clone(), |set, answers| {
+                    set.intersection(answers).copied().collect()
+                })
+        })
+        .map(|set| set.len())
+        .sum())
+}
+
 struct Group {
     answers: Vec<HashSet<u8>>,
 }
@@ -68,4 +87,14 @@ mod tests {
     fn part1_real() {
         assert_eq!(part1(INPUT).unwrap(), 6382);
     }
+
+    #[test]
+    fn part2_provided() {
+        assert_eq!(part2(PROVIDED).unwrap(), 6);
+    }
+
+    #[test]
+    fn part2_real() {
+        assert_eq!(part2(INPUT).unwrap(), 3197);
+    }
 }