Update read-n-characters-given-read4.cpp

kamyu104 · web-flow · commit 405f9cf90a38 · 2018-08-04T14:21:10.000+08:00
diff --git a/C++/read-n-characters-given-read4.cpp b/C++/read-n-characters-given-read4.cpp
@@ -13,10 +13,14 @@ class Solution {
     int read(char *buf, int n) {
         int read_bytes = 0;
         char buffer[4];
-        while (int size = read4(buffer)) {
-            size = min(size, n - read_bytes);
-            for (int i = 0; i < size; ++i) {
-                buf[read_bytes++] = buffer[i];
+        for (int i = 0; i <= n / 4; ++i) {
+            if (int size = read4(buffer)) {
+                size = min(size, n - read_bytes);
+                for (int j = 0; j < size; ++j) {
+                    buf[read_bytes++] = buffer[j];
+                }
+            } else {
+                break;
             }
         }
         return read_bytes;
