Codeforces (431A) Problem Solution

 A - Black Square

Problem Link: Codeforces 431A

• C++
• Python3
• JAVA

#include <iostream>
#include <vector>
#include <bits/stdc++.h>
#define ll long long
 
 
using namespace std;
 
 
int main()
{
 
    int a, b , c, d, som = 0;
    cin >> a >> b >> c >> d;
    string z;
    cin >> z;
 
    for(int i = 0; i < z.length();i++)
    {
        if(z[i] == '1')
            som+=a;
        if(z[i] == '2')
            som+=b;
        if(z[i] == '3')
            som+=c;
        if(z[i] == '4')
            som+=d;
 
    }
 
    cout << som;
     return 0;
}

c=list(map(int,input().split()))
s=input()
sumc=0
for i in range(len(s)):
    if s[i]=='1':
        sumc=sumc+c[0]
    if s[i]=='2':
        sumc=sumc+c[1]
    if s[i]=='3':
        sumc=sumc+c[2]
    if s[i]=='4':
        sumc=sumc+c[3]
        
print(sumc)    

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Scanner;
 
//  gravity flip problem
public class App {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int a1 = sc.nextInt(), calories = 0;
        int a2 = sc.nextInt();
        int a3 = sc.nextInt();
        int a4 = sc.nextInt();
        sc.nextLine();
        String[] values = sc.nextLine().split("");
        for (String i : values) {
            switch (i) {
                case "1":
                    calories += a1;
                    break;
                case "2":
                    calories += a2;
                    break;
                case "3":
                    calories += a3;
                    break;
                case "4":
                    calories += a4;
                    break;
            }
        }
        System.out.println(calories);
        sc.close();
    }
}