Solved two problem in codechef

1 files changed, 26 insertions, 0 deletions

diff --git a/codechef/chefInVaccinationQueue.cpp b/codechef/chefInVaccinationQueue.cpp
new file mode 100644
index 0000000..eb2bfdd
--- /dev/null
+++ b/codechef/chefInVaccinationQueue.cpp
@@ -0,0 +1,26 @@
+#include <bits/stdc++.h>
+
+using namespace std;
+
+int main() {
+  ios_base::sync_with_stdio(false);
+  cin.tie(NULL);
+  int tt;
+  cin >> tt;
+  while (tt--) {
+    int n, p, x, y;
+    cin >> n >> p >> x >> y;
+    vector<int> v(n);
+    int ans = 0;
+    for (int &x : v)
+      cin >> x;
+    for (int i = 0; i < p; i++) {
+      if (v[i] == 0) {
+        ans += x;
+      } else {
+        ans += y;
+      }
+    }
+    cout << ans << endl;
+  }
+}