Initial commit.HEADmain

1 files changed, 25 insertions, 0 deletions

diff --git a/top-interview-questions/easy/array/07_plus_one.cc b/top-interview-questions/easy/array/07_plus_one.cc
new file mode 100644
index 0000000..05c7b21
--- /dev/null
+++ b/top-interview-questions/easy/array/07_plus_one.cc
@@ -0,0 +1,25 @@
+class Solution {
+public:
+    vector<int> plusOne(vector<int>& digits) {
+        int carry = 1;
+
+        for (int i = digits.size()-1; (i >= 0) && (carry == 1); --i) {
+            const int sum = digits[i] + carry;
+            digits[i] = sum % 10;
+            carry     = (sum >= 10) ? 1 : 0;
+        }
+
+        if (carry == 1) { // Overflow.
+            digits.resize(digits.size() + 1);
+
+            // Shift right.
+            for (size_t i = digits.size()-1; i > 0; --i) {
+                digits[i] = digits[i-1];
+            }
+
+            digits[0] = 1;
+        }
+
+        return digits;
+    }
+};