Fix copy_n to increment only n-1 times for an input iterator. This works much better with std::istream_iterator<int>(std::cin). Credit: Matan Nassau.

llvm-svn: 126581

1 files changed, 10 insertions, 1 deletions

diff --git a/libcxx/include/algorithm b/libcxx/include/algorithm
index d91c57c1109..4909221bb09 100644
--- a/libcxx/include/algorithm
+++ b/libcxx/include/algorithm
@@ -1559,8 +1559,17 @@ typename enable_if
 >::type
 copy_n(_InputIterator __first, _Size __n, _OutputIterator __result)
 {
-    for (; __n > 0; --__n, ++__first, ++__result)
+    if (__n > 0)
+    {
         *__result = *__first;
+        ++__result;
+        for (--__n; __n > 0; --__n)
+        {
+            ++__first;
+            *__result = *__first;
+            ++__result;
+        }
+    }
     return __result;
 }