aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--water_bottles.c9
1 files changed, 1 insertions, 8 deletions
diff --git a/water_bottles.c b/water_bottles.c
index 20a0a2d..6f11346 100644
--- a/water_bottles.c
+++ b/water_bottles.c
@@ -10,14 +10,7 @@
*/
int numWaterBottles(int numBottles, int numExchange) {
- int ans = numBottles;
- while (numBottles >= numExchange) {
- int rem = numBottles % numExchange;
- numBottles /= numExchange;
- ans += numBottles;
- numBottles += rem;
- }
- return ans;
+ return numBottles + (numBottles - 1) / (numExchange - 1);
}
int main() {