Fix array subtraction

phase · Jul 26, 2015 · 11e3afe · 11e3afe
1 parent 6970231
commit 11e3afe
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/xyz/jadonfowler/o/O.java b/src/xyz/jadonfowler/o/O.java
@@ -289,12 +289,21 @@ else if (a instanceof String || b instanceof String) {
         else if (c == '-') {
             if (stacks[sid].peek() instanceof ArrayList<?>) {
                 double ans = 0;
+                boolean set = false;
                 for (Object o : (ArrayList<Object>) stacks[sid].pop()) {
                     if (o instanceof Integer) {
-                        ans -= (int) o;
+                        if (!set) {
+                            ans = (int) o;
+                            set = true;
+                        }
+                        else ans -= (int) o;
                     }
                     else if (o instanceof Double) {
-                        ans -= (double) o;
+                        if (!set) {
+                            ans = (double) o;
+                            set = true;
+                        }
+                        else ans -= (double) o;
                     }
                 }
                 stacks[sid].push(ans);