From a57559f521668a913503d924dfbe42cbbb6c9bbf Mon Sep 17 00:00:00 2001
From: Jason Yuen <jason_yuen2007@hotmail.com>
Date: Sun, 18 Jun 2023 09:07:58 +0000
Subject: [PATCH] feat: port Archive.Imo.Imo2019Q1 (#5211)

---
 Archive.lean               |  1 +
 Archive/Imo/Imo2019Q1.lean | 56 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 57 insertions(+)
 create mode 100644 Archive/Imo/Imo2019Q1.lean

diff --git a/Archive.lean b/Archive.lean
index 2872b4eb45204..88f92e1318f83 100644
--- a/Archive.lean
+++ b/Archive.lean
@@ -18,6 +18,7 @@ import Archive.Imo.Imo2008Q4
 import Archive.Imo.Imo2011Q3
 import Archive.Imo.Imo2011Q5
 import Archive.Imo.Imo2013Q1
+import Archive.Imo.Imo2019Q1
 import Archive.Imo.Imo2019Q4
 import Archive.Imo.Imo2020Q2
 import Archive.Wiedijk100Theorems.AreaOfACircle
diff --git a/Archive/Imo/Imo2019Q1.lean b/Archive/Imo/Imo2019Q1.lean
new file mode 100644
index 0000000000000..1e4aedc70069f
--- /dev/null
+++ b/Archive/Imo/Imo2019Q1.lean
@@ -0,0 +1,56 @@
+/-
+Copyright (c) 2020 Kevin Buzzard. All rights reserved.
+Released under Apache 2.0 license as described in the file LICENSE.
+Authors: Kevin Buzzard
+
+! This file was ported from Lean 3 source module imo.imo2019_q1
+! leanprover-community/mathlib commit 308826471968962c6b59c7ff82a22757386603e3
+! Please do not edit these lines, except to modify the commit id
+! if you have ported upstream changes.
+-/
+import Mathlib.Tactic.Linarith
+
+/-!
+# IMO 2019 Q1
+
+Determine all functions `f : ℤ → ℤ` such that, for all integers `a` and `b`,
+`f(2a) + 2f(b) = f(f(a+b))`.
+
+The desired theorem is that either:
+  - `f = fun _ ↦ 0`
+  - `∃ c, f = fun x ↦ 2 * x + c`
+
+Note that there is a much more compact proof of this fact in Isabelle/HOL
+  - http://downthetypehole.de/paste/4YbGgqb4
+-/
+
+
+theorem imo2019_q1 (f : ℤ → ℤ) :
+    (∀ a b : ℤ, f (2 * a) + 2 * f b = f (f (a + b))) ↔ f = 0 ∨ ∃ c, f = fun x => 2 * x + c := by
+  constructor; swap
+  -- easy way: f(x)=0 and f(x)=2x+c work.
+  · rintro (rfl | ⟨c, rfl⟩) <;> intros <;> simp only [Pi.zero_apply]; ring
+  -- hard way.
+  intro hf
+  -- functional equation
+  -- Using `h` for `(0, b)` and `(-1, b + 1)`, we get `f (b + 1) = f b + m`
+  obtain ⟨m, H⟩ : ∃ m, ∀ b, f (b + 1) = f b + m := by
+    refine' ⟨(f 0 - f (-2)) / 2, fun b => _⟩
+    refine' sub_eq_iff_eq_add'.1 (Int.eq_ediv_of_mul_eq_right two_ne_zero _)
+    have h1 : f 0 + 2 * f b = f (f b) := by simpa using hf 0 b
+    have h2 : f (-2) + 2 * f (b + 1) = f (f b) := by simpa using hf (-1) (b + 1)
+    linarith
+  -- Hence, `f` is an affine map, `f b = f 0 + m * b`
+  obtain ⟨c, H⟩ : ∃ c, ∀ b, f b = c + m * b := by
+    refine' ⟨f 0, fun b => _⟩
+    induction' b using Int.induction_on with b ihb b ihb
+    · simp
+    · simp [H, ihb, mul_add, add_assoc]
+    · rw [← sub_eq_of_eq_add (H _)]
+      simp [ihb]; ring
+  -- Now use `hf 0 0` and `hf 0 1` to show that `m ∈ {0, 2}`
+  have H3 : 2 * c = m * c := by simpa [H, mul_add] using hf 0 0
+  obtain rfl | rfl : 2 = m ∨ m = 0 := by simpa [H, mul_add, H3] using hf 0 1
+  · right; use c; ext b; simp [H, add_comm]
+  · left; ext b; simpa [H, two_ne_zero] using H3
+#align imo2019_q1 imo2019_q1