From 239b9e82ea45639725e511d957d78a33d329c453 Mon Sep 17 00:00:00 2001
From: Pulkit Agarwal <pulkitagarwal03@gmail.com>
Date: Sat, 7 Oct 2017 19:08:06 +0530
Subject: [PATCH] Add zco14001.cpp (#5)

---
 codechef/zco14001.cpp | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 codechef/zco14001.cpp

diff --git a/codechef/zco14001.cpp b/codechef/zco14001.cpp
new file mode 100644
index 0000000..4df5601
--- /dev/null
+++ b/codechef/zco14001.cpp
@@ -0,0 +1,28 @@
+#include<iostream>
+ 
+using namespace std;
+ 
+int main()
+{
+    int n,h;
+    cin>>n>>h;
+    int stack[n];
+    for(int i=0;i<n;++i)
+        cin>>stack[i];
+    int com,crane=0,loaded=0;
+    do
+    {
+        cin>>com;
+        switch(com)
+        {
+            case 1: if(crane != 0) crane--;break;
+            case 2: if(crane != (n-1)) crane++;break;
+            case 3: if(loaded == 0 && stack[crane] != 0){stack[crane]--; loaded = 1;}break;
+            case 4: if(loaded == 1 && stack[crane] != (h)){stack[crane]++; loaded = 0;}break;
+        }
+    }while(com!=0);
+    for(int i=0; i<n;++i)
+    {
+        cout<<stack[i]<<" ";
+    }
+}