From 9021452a20ffb34e7cff2407643861d82e94387d Mon Sep 17 00:00:00 2001
From: Lam Haoyin <root@eatswap.org>
Date: Sun, 27 Mar 2022 10:46:27 +0800
Subject: [PATCH] feat: simplify 220327-CN [cpp]

---
 cpp/2203/220327-CN.cpp | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

diff --git a/cpp/2203/220327-CN.cpp b/cpp/2203/220327-CN.cpp
index 4b78fe3..14fec72 100644
--- a/cpp/2203/220327-CN.cpp
+++ b/cpp/2203/220327-CN.cpp
@@ -19,15 +19,10 @@ public:
 			return {};
 		std::vector<int> ret(n, rem / n);
 		rem = rem - rem / n * n;
-		for (int& i : ret) {
-			if (rem <= 0)
-				break;
-			if (rem < 6 - i) {
-				i += rem;
-				break;
-			}
-			rem -= 6 - i;
-			i = 6;
+		for (int i = 0; i < n && rem > 0; ++i) {
+			int delta = std::min(rem, 6 - ret[i]);
+			rem -= delta;
+			ret[i] += delta;
 		}
 		return ret;
 	}