Problem:


Score : 600 points

Problem Statement

There is a game that involves three variables, denoted A, B, and C.

As the game progresses, there will be N events where you are asked to make a choice.
Each of these choices is represented by a string s_i. If s_i is AB, you must add 1 to A or B then subtract 1 from the other; if s_i is AC, you must add 1 to A or C then subtract 1 from the other; if s_i is BC, you must add 1 to B or C then subtract 1 from the other.

After each choice, none of A, B, and C should be negative.

Determine whether it is possible to make N choices under this condition. If it is possible, also give one such way to make the choices.

Constraints

1 \leq N \leq 10^5

0 \leq A,B,C \leq 10^9

N, A, B, C are integers.

s_i is AB, AC, or BC.

Input

Input is given from Standard Input in the following format:

N A B C
s_1
s_2
:
s_N

Output

If it is possible to make N choices under the condition, print Yes; otherwise, print No.

Also, in the former case, show one such way to make the choices in the subsequent N lines. The (i+1)-th line should contain the name of the variable (A, B, or C) to which you add 1 in the i-th choice.

Sample Input 1

2 1 3 0
AB
AC

Sample Output 1

Yes
A
C

You can successfully make two choices, as follows:

In the first choice, add 1 to A and subtract 1 from B. A becomes 2, and B becomes 2.

In the second choice, add 1 to C and subtract 1 from A. C becomes 1, and A becomes 1.

Sample Input 2

3 1 0 0
AB
BC
AB

Sample Output 2

No

Sample Input 3

1 0 9 0
AC

Sample Output 3

No

Sample Input 4

8 6 9 1
AC
BC
AB
BC
AC
BC
AB
AB

Sample Output 4

Yes
C
B
B
C
C
B
A
A



Sample input:
2 1 3 0
AB
AC


Sample output:
Yes
A
C

Submission status: Wrong Answer

Solution code:
import sys
sys.setrecursionlimit(10**8)
def ii(): return int(sys.stdin.readline())
def mi(): return map(int, sys.stdin.readline().split())
def li(): return list(map(int, sys.stdin.readline().split()))
def li2(N): return [list(map(int, sys.stdin.readline().split())) for i in range(N)]
def dp2(ini, i, j): return [[ini]*i for i2 in range(j)]
def dp3(ini, i, j, k): return [[[ini]*i for i2 in range(j)] for i3 in range(k)]
import math
#import bisect #bisect.bisect_left(B, a)
#from collections import defaultdict #d = defaultdict(int) d[key] += value
#from collections import Counter # a = Counter(A).most_common()
#from itertools import accumulate #list(accumulate(A))

N, *D = mi()
S = [sys.stdin.readline()[:2] for _ in range(N)]

def cv(m):
    if m == 'A':
        return 0
    if m == 'B':
        return 1
    if m == 'C':
        return 2


yoso = ['A', 'B', 'C']
ans = []
for i in range(N):
    s, t = cv(S[i][0]), cv(S[i][1])
    if D[s] == D[t] == 0:
            print('No')
            exit()
    if sum(D) >= 3:
        if D[s] >= D[t]:
            D[s] -= 1
            D[t] += 1
            ans.append(S[i][1])
        else:
            D[s] += 1
            D[t] -= 1
            ans.append(S[i][0])
    elif sum(D) == 2:
        if D[s] == 2:
            D[s] -= 1
            D[t] += 1
            ans.append(S[i][1])
        elif D[t] == 2:
            D[t] -= 1
            D[s] += 1
            ans.append(S[i][0])
        elif D[s] == D[t] == 1:
            if i+1 < N:
                if S[i] == S[i+1]:
                    D[s] = 1 - D[s]
                    D[t] = 1 - D[t]
                    if D[s] == 1:
                        ans.append(S[i][0])
                    else:
                        ans.append(S[i][1])
                if cv(S[i][0]) == s or cv(S[i][1]) == S:
                    D[s] += 1
                    ans.append(S[i][0])
                else:
                    D[t] += 1
                    ans.append(S[i][1])
            else:
                ans.append(S[i][0])
        else:
            D[s] = 1 - D[s]
            D[t] = 1 - D[t]
            if D[s] == 1:
                ans.append(S[i][0])
            else:
                ans.append(S[i][1])
    else:
        if D[s] == 1 or D[t] == 1:
            D[s] = 1 - D[s]
            D[t] = 1 - D[t]
            if D[s] == 1:
                ans.append(S[i][0])
            else:
                ans.append(S[i][1])

print('Yes')
for m in ans:
    print(m)

Submission status: Wrong Answer

Solution code:
import sys
sys.setrecursionlimit(10**8)
def ii(): return int(sys.stdin.readline())
def mi(): return map(int, sys.stdin.readline().split())
def li(): return list(map(int, sys.stdin.readline().split()))
def li2(N): return [list(map(int, sys.stdin.readline().split())) for i in range(N)]
def dp2(ini, i, j): return [[ini]*i for i2 in range(j)]
def dp3(ini, i, j, k): return [[[ini]*i for i2 in range(j)] for i3 in range(k)]
import math
#import bisect #bisect.bisect_left(B, a)
#from collections import defaultdict #d = defaultdict(int) d[key] += value
#from collections import Counter # a = Counter(A).most_common()
#from itertools import accumulate #list(accumulate(A))

N, *D = mi()
S = [sys.stdin.readline()[:2] for _ in range(N)]

def cv(m):
    if m == 'A':
        return 0
    if m == 'B':
        return 1
    if m == 'C':
        return 2


yoso = ['A', 'B', 'C']
ans = []
for i in range(N):
    s, t = cv(S[i][0]), cv(S[i][1])
    if D[s] == D[t] == 0:
            print('No')
            exit()
    if sum(D) > 3:
        if D[s] >= D[t]:
            D[s] -= 1
            D[t] += 1
            ans.append(S[i][1])
        else:
            D[s] += 1
            D[t] -= 1
            ans.append(S[i][0])
    elif sum(D) == 2:
        if D[s] == 2:
            D[s] -= 1
            D[t] += 1
            ans.append(S[i][1])
        elif D[t] == 2:
            D[t] -= 1
            D[s] += 1
            ans.append(S[i][0])
        elif D[s] == D[t] == 1:
            if i+1 < N:
                if S[i] == S[i+1]:
                    D[s] = 1 - D[s]
                    D[t] = 1 - D[t]
                    if D[s] == 1:
                        ans.append(S[i][0])
                    else:
                        ans.append(S[i][1])
                if cv(S[i][0]) == s or cv(S[i][1]) == S:
                    D[s] += 1
                    ans.append(S[i][0])
                else:
                    D[t] += 1
                    ans.append(S[i][1])
            else:
                ans.append(S[i][0])
        else:
            D[s] = 1 - D[s]
            D[t] = 1 - D[t]
            if D[s] == 1:
                ans.append(S[i][0])
            else:
                ans.append(S[i][1])
    else:
        if D[s] == 1 or D[t] == 1:
            D[s] = 1 - D[s]
            D[t] = 1 - D[t]
            if D[s] == 1:
                ans.append(S[i][0])
            else:
                ans.append(S[i][1])

print('Yes')
for m in ans:
    print(m)

Submission status: Wrong Answer

Solution code:
import sys
input = lambda:sys.stdin.readline().strip()
n,a,b,c = map(int,input().split())
ok = True
ans = []
for i in range(n):
    s = input()
    if s=='AB':
        if a>b:
            a-=1
            b+=1
            ans.append('B')
        else:
            b-=1
            a+=1
            ans.append('A')
    elif s=='AC':
        if a>c:
            a-=1
            c+=1
            ans.append('C')
        else:
            c-=1
            a+=1
            ans.append('A')
    else:
        if b>c:
            b-=1
            c+=1
            ans.append('C')
        else:
            c-=1
            b+=1
            ans.append('B')
    if a<0 or b<0 or c<0:
        print("No")
        ok = False
        break
if ok:
    print("Yes")
    for i in ans:
        print(i)

Submission status: Wrong Answer

Solution code:
n,a,b,c = map(int,input().split())
clear=1
ans=""

if a+b+c!=2:
    for _ in range(n):
        s=input()
        if s=="AB":
            if a>b:
                a-=1
                b+=1
                ans+="B"
            else:
                a+=1
                b-=1
                ans+="A"
        elif s=="BC":
            if b>c:
                b-=1
                c+=1
                ans+="C"
            else:
                b+=1
                c-=1
                ans+="B"
        else:
            if c>a:
                c-=1
                a+=1
                ans+="A"
            else:
                a-=1
                c+=1
                ans+="C"
        if (a<0) or (b<0) or (c<0):
            clear=0
else:
    memo=""
    for _ in range(n):
        s=input()
        if s=="AB":
            if (a==0) and (b==0) and (memo!=""):
                piece=memo.strip(ans[-1])
                if piece=="A":
                    a=0
                    b=1
                    c-=1
                    ans=ans[:-1]+"AB"
                else:
                    a=1
                    b=0
                    c-=1
                    ans=ans[:-1]+"BA"
            elif a>b:
                a-=1
                b+=1
                ans+="B"
            else:
                a+=1
                b-=1
                ans+="A"
                    
                
        elif s=="BC":
            if (b==0) and (c==0) and (memo!=""):
                piece=memo.strip(ans[-1])
                if piece=="B":
                    b=0
                    c=1
                    a-=1
                    ans=ans[:-1]+"BC"
                else:
                    a=1
                    b=0
                    c-=1
                    ans=ans[:-1]+"CB"
            elif b>c:
                b-=1
                c+=1
                ans+="C"
            else:
                b+=1
                c-=1
                ans+="B"
        else:
            if (a==0) and (c==0) and (memo!=""):
                piece=memo.strip(ans[-1])
                if piece=="A":
                    a=0
                    c=1
                    b-=1
                    ans=ans[:-1]+"AC"
                else:
                    a=1
                    c=0
                    b-=1
                    ans=ans[:-1]+"CA"
            elif c>a:
                c-=1
                a+=1
                ans+="A"
            else:
                a-=1
                c+=1
                ans+="C"
        if (a<0) or (b<0) or (c<0):
            clear=0
        memo=s
        
if clear==1:
    print("Yes")
    for i in ans:
        print(i)
else:
    print("No")

Submission status: Wrong Answer

Solution code:
if __name__=='__main__':
    n, a, b, c = map(int, input().split())
    res = []
    flag = 0
    s_lst = []
    for i in range(n):
        s = input()
        s_lst.append(s)
        
    for i, s in enumerate(s_lst):
        if a + b + c == 0:
            print('No')
            flag = 1
            break
        elif a + b + c == 1:
            if s == 'AB':
                if a == 1:
                    a -= 1
                    b += 1
                    res.append('B')
                elif b == 1:
                    b -= 1
                    a += 1
                    res.append('A')
                else:
                    print('No')
                    flag = 1
                    break
            elif s == 'AC':
                if a == 1:
                    a -= 1
                    c += 1
                    res.append('C')
                elif b == 1:
                    print('No')
                    flag = 1
                    break
                else:
                    c -= 1
                    a += 1
                    res.append('A')
            else:
                if a == 1:
                    print('No')
                    flag = 1
                    break
                elif b == 1:
                    b -= 1
                    c += 1
                    res.append('C')
                else:
                    c -= 1
                    b += 1
                    res.append('B')
        elif a + b + c >= 2:
            if s == 'AB':
                if b == 0:
                    if a == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        a -= 1
                        b += 1
                        res.append('B')
                else:
                    if a == 0:
                        b -= 1
                        a += 1
                        res.append('A')                        
                    else:
                        if a + b + c == 2 and i != (n-1):
                            if a == 1 and b == 1:
                                if s_lst[i+1] == 'AC':
                                    a += 1
                                    b -= 1
                                    res.append('A')
                                elif s_lst[i+1] == 'BC':
                                    a -= 1
                                    b += 1
                                    res.append('B')
                            else:
                                if a > b:
                                    a -= 1
                                    b += 1
                                    res.append('B')
                                else:
                                    b -= 1
                                    a += 1
                                    res.append('A')                        
                        else:
                            if a > b:
                                a -= 1
                                b += 1
                                res.append('B')
                            else:
                                b -= 1
                                a += 1
                                res.append('A')                            
            elif s == 'AC':
                if c == 0:
                    if a == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        a -= 1
                        c += 1
                        res.append('C')
                else:
                    if a == 0:
                        c -= 1
                        a += 1
                        res.append('A')                        
                    else:
                        if a + b + c == 2 and i != (n-1):
                            if a == 1 and c == 1:
                                if s_lst[i+1] == 'AB':
                                    a += 1
                                    c -= 1
                                    res.append('A')
                                elif s_lst[i+1] == 'BC':
                                    a -= 1
                                    c += 1
                                    res.append('C')
                            else:
                                if a > c:
                                    a -= 1
                                    c += 1
                                    res.append('C')
                                else:
                                    c -= 1
                                    a += 1
                                    res.append('A')                        
                        else:
                            if a > c:
                                a -= 1
                                c += 1
                                res.append('C')
                            else:
                                c -= 1
                                a += 1
                                res.append('A') 
            else:
                if c == 0:
                    if b == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        b -= 1
                        c += 1
                        res.append('C')
                else:
                    if b == 0:
                        c -= 1
                        b += 1
                        res.append('B')                        
                    else:
                        if a + b + c == 2 and i != (n-1):
                            if b == 1 and c == 1:
                                if s_lst[i+1] == 'AB':
                                    b += 1
                                    c -= 1
                                    res.append('A')
                                elif s_lst[i+1] == 'AC':
                                    b -= 1
                                    c += 1
                                    res.append('B')
                            else:
                                if b > c:
                                    b -= 1
                                    c += 1
                                    res.append('C')
                                else:
                                    c -= 1
                                    b += 1
                                    res.append('B')                        
                        else:
                            if b > c:
                                b -= 1
                                c += 1
                                res.append('C')
                            else:
                                c -= 1
                                b += 1
                                res.append('B')   
    if flag == 0:
        print('Yes')
        for elem in res:
            print(elem)

Submission status: Wrong Answer

Solution code:
import numpy as np
N,a,b,c = map(int, input().split())
queries = [input() for _ in range(N)]
point = np.array([a,b,c])
result = np.array([''])
for n,s in enumerate(queries):
    if s =='AB':
        exp1 = np.array([1,-1,0]) + point
        exp2 = np.array([-1,1,0]) + point
        expr1 = result.astype(object) + np.array(['A'],dtype=object) 
        expr2 = result.astype(object) + np.array(['B'],dtype=object) 
    elif s =='AC':
        exp1 = np.array([1,0,-1]) + point
        exp2 = np.array([-1,0,1]) + point
        expr1 = result.astype(object) + np.array(['A'],dtype=object) 
        expr2 = result.astype(object) + np.array(['C'],dtype=object) 
    elif s =='BC':
        exp1 = np.array([0,1,-1]) + point
        exp2 = np.array([0,-1,1]) + point
        expr1 = result.astype(object) + np.array(['B'],dtype=object) 
        expr2 = result.astype(object) + np.array(['C'],dtype=object)
    else:
        exit()
    point = np.append(exp1, exp2, axis=0).reshape(2**(n+1),3)
    result = np.append(expr1,expr2,axis=0).reshape(2**(n+1),1)
if True in list(np.all(point > 0, axis=1)):
    indexN = list(np.all(point > 0, axis=1)).index(True)
    print("Yes")
    for i in range(len(result[indexN,0])):
        print(result[indexN,0][i])
else:
    print("No")

Submission status: Wrong Answer

Solution code:
if __name__=='__main__':
    n, a, b, c = map(int, input().split())
    res = []
    flag = 0
    s_lst = []
    for i in range(n):
        s = input()
        s_lst.append(s)
        
    for i, s in enumerate(s_lst):
        if a + b + c == 0:
            print('No')
            flag = 1
            break
        elif a + b + c == 1:
            if s == 'AB':
                if a == 0:
                    if b == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        b -= 1
                        a += 1
                        res.append('A')
                else:
                    a -= 1
                    b += 1
                    res.append('B')
            elif s == 'AC':
                if a == 0:
                    if c == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        c -= 1
                        a += 1
                        res.append('A')
                else:
                    a -= 1
                    c += 1
                    res.append('C')
            else:
                if c == 0:
                    if b == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        b -= 1
                        c += 1
                        res.append('C')
                else:
                    c -= 1
                    b += 1
                    res.append('B')
        elif a + b + c >= 2:
            if s == 'AB':
                if b == 0:
                    if a == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        a -= 1
                        b += 1
                        res.append('B')
                else:
                    if a + b + c == 2 and i != (n-1):
                        if a == 1 and b == 1:
                            if s_lst[i+1] == 'AC':
                                a += 1
                                b -= 1
                                res.append('A')
                            elif s_lst[i+1] == 'BC':
                                a -= 1
                                b += 1
                                res.append('B')
                        else:
                            if a == 0:
                                a += 1
                                b -= 1
                                res.append('A')
                            else:
                                a -= 1
                                b += 1
                                res.append('B')                           
                    else:
                        if a > b:
                            a -= 1
                            b += 1
                            res.append('B')
                        else:
                            b -= 1
                            a += 1
                            res.append('A')                            
            elif s == 'AC':
                if c == 0:
                    if a == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        a -= 1
                        c += 1
                        res.append('C')
                else:
                    if a + b + c == 2 and i != (n-1):
                        if a == 1 and c == 1:
                            if s_lst[i+1] == 'AB':
                                a += 1
                                c -= 1
                                res.append('A')
                            elif s_lst[i+1] == 'BC':
                                a -= 1
                                c += 1
                                res.append('C')
                        else:
                            if a == 0:
                                a += 1
                                c -= 1
                                res.append('A')
                            else:
                                a -= 1
                                c += 1
                                res.append('C')  
                    else:
                        if a > c:
                            a -= 1
                            c += 1
                            res.append('C')
                        else:
                            c -= 1
                            a += 1
                            res.append('A')  
            else:
                if c == 0:
                    if b == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        b -= 1
                        c += 1
                        res.append('C')
                else:
                    if a + b + c == 2 and i != (n-1):
                        if b == 1 and c == 1:
                            if s_lst[i+1] == 'AC':
                                b -= 1
                                c += 1
                                res.append('C')
                            elif s_lst[i+1] == 'AB':
                                b += 1
                                c -= 1
                                res.append('B')
                        else:
                            if b == 0:
                                b += 1
                                c -= 1
                                res.append('B')
                            else:
                                b -= 1
                                c += 1
                                res.append('C')  
                    else:
                        if b > c:
                            b -= 1
                            c += 1
                            res.append('C')
                        else:
                            c -= 1
                            b += 1
                            res.append('B')  
    if flag == 0:
        print('Yes')
        for elem in res:
            print(elem)

Submission status: Wrong Answer

Solution code:
if __name__=='__main__':
    n, a, b, c = map(int, input().split())
    res = []
    flag = 0
    s_lst = []
    for i in range(n):
        s = input()
        s_lst.append(s)
        
    for i, s in enumerate(s_lst):
        if a + b + c == 0:
            print('No')
            flag = 1
            break
        elif a + b + c == 1:
            if s == 'AB':
                if a == 0:
                    if b == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        b -= 1
                        a += 1
                        res.append('A')
                else:
                    a -= 1
                    b += 1
                    res.append('B')
            elif s == 'AC':
                if a == 0:
                    if c == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        c -= 1
                        a += 1
                        res.append('A')
                else:
                    a -= 1
                    c += 1
                    res.append('C')
            else:
                if c == 0:
                    if b == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        b -= 1
                        c += 1
                        res.append('C')
                else:
                    c -= 1
                    b += 1
                    res.append('B')
        elif a + b + c > 1:
            if s == 'AB':
                if b == 0:
                    if a == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        a -= 1
                        b += 1
                        res.append('B')
                else:
                    if a + b + c == 2 and i != (n-1):
                        if a == 1 and b == 1:
                            if s_lst[i+1] == 'AC':
                                a += 1
                                b -= 1
                                res.append('A')
                            elif s_lst[i+1] == 'BC':
                                a -= 1
                                b += 1
                                res.append('B')
                    else:
                        if a > b:
                            a -= 1
                            b += 1
                            res.append('B')
                        else:
                            b -= 1
                            a += 1
                            res.append('A')                            
            elif s == 'AC':
                if c == 0:
                    if a == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        a -= 1
                        c += 1
                        res.append('C')
                else:
                    if a + b + c == 2 and i != (n-1):
                        if a == 1 and c == 1:
                            if s_lst[i+1] == 'AB':
                                a += 1
                                c -= 1
                                res.append('A')
                            elif s_lst[i+1] == 'BC':
                                a -= 1
                                c += 1
                                res.append('C')
                    else:
                        if a > c:
                            a -= 1
                            c += 1
                            res.append('C')
                        else:
                            c -= 1
                            a += 1
                            res.append('A')  
            else:
                if c == 0:
                    if b == 0:
                        print('No')
                        flag = 1
                        break
                    else:
                        b -= 1
                        c += 1
                        res.append('C')
                else:
                    if a + b + c == 2 and i != (n-1):
                        if b == 1 and c == 1:
                            if s_lst[i+1] == 'AC':
                                b -= 1
                                c += 1
                                res.append('C')
                            elif s_lst[i+1] == 'AB':
                                b += 1
                                c -= 1
                                res.append('B')
                    else:
                        if b > c:
                            b -= 1
                            c += 1
                            res.append('C')
                        else:
                            c -= 1
                            b += 1
                            res.append('B')  
    if flag == 0:
        print('Yes')
        for elem in res:
            print(elem)

Submission status: Wrong Answer

Solution code:
print('Yes')
res = ['B','A']
for i in res:
    print(i)

Submission status: Wrong Answer

Solution code:
print('Yes')
res = ['A','C']
for i in res:
    print(i)