★BF Upper-Intermediate I - 2 Solved★

★ 2503 숫자 야구 ★

 

import sys
from itertools import permutations
input=sys.stdin.readline

N=int(input())
pos=list(permutations(['1','2','3','4','5','6','7','8','9'],3))
idxs=[n for n in range(len(pos))]

def get_strike_ball(a,b):
    strike=0
    for x,y in zip(a,b):
        if x==y: strike+=1
    return (str(strike),str(len(a+b)-len(set(a+b))-strike))

for _ in range(N):
    num,S,B=input().split()
    didxs=[]
    for i in idxs:
        posi=''.join(list(pos[i]))
        if get_strike_ball(list(posi),list(num)) != (S,B): didxs.append(i)
    for di in didxs: idxs.remove(di)
print(len(idxs))

 

🌞 123과 같이 9개의 자연수 중 순서 있는 서로 다른 3개의 자연수로 이루어진 쌍의 개수 9P3 permutations()

🌞 strike / ball 개수 count

: 먼저 strike 개수 세고, 전체 개수 - 중복 없는 개수 - strike 개수를 ball 개수

 

🌞 permutations() 결과 pos list brute-force(범위 작아서 brute-force 가능) → join() / list()로 type 통일 뒤 원래의 (S,B)와 비교 → 다르다면 별도의 삭제 idx 모음 didx update(그 자리에서 remove하면 오류 발생: 해결 방법은 별도의 리스트 만들어 한 번에 모은 뒤 따로 for loop 돌려 remove) → 이후 didx for loop 돌려 바로 idx remove() → 최종 남은 idx list 원소 수 출력

---

★ 3085 사탕 게임 ★

 

#potential swap included in an array
import sys,copy
input=sys.stdin.readline

def get_consecutive_max(arr):
    ans,cur=1,1
    for i in range(1,len(arr)):
        if arr[i]==arr[i-1]:
            cur+=1
        else:
            ans=max(ans,cur)
            cur=1
    return max(ans,cur)

def bomboni(arr):
    cur=0
    n=len(arr)
    #row-wise
    for i in range(0,n):
        cur=max(cur,get_consecutive_max(arr[i]))
    #column-wise
    for i in range(n):
        column_arr=[]
        for j in range(n):
            column_arr.append(arr[j][i])
        cur=max(cur,get_consecutive_max(column_arr))
    return cur

N=int(input())
ans,arr=0,[]
for _ in range(N): arr.append(list(input().rstrip())) #made an array
ans=0
for i in range(N):
    for j in range(N):
        if j<=(N-2) and arr[i][j] != arr[i][j+1]: #right dir
            arr2=copy.deepcopy(arr)
            arr2[i][j], arr2[i][j+1] = arr2[i][j+1], arr2[i][j]
            ans=max(ans,bomboni(arr2))
        if i<=(N-2) and arr[i][j] != arr[i+1][j]: #down dir
            arr2=copy.deepcopy(arr)
            arr2[i][j], arr2[i+1][j] = arr2[i+1][j], arr2[i][j]
            ans=max(ans,bomboni(arr2))    
print(ans)

 

① array traversal 과정에서 swap되는 direction → right direction / downward direction

: 이미 만들어진 array가 아닌 새로운 arr2 만들어서(deepcopy() 사용) bomboni() 진행

 

② bomboni(arr)

:  row-wise / column-wise 방향 traversal

 

③ get_consecutive(arr)

: 주어진 1차원 array에서 가장 길게 연속하는 배열 길이 출력

: ans 변수 누적 update(cur 변수 iterate) → 마지막 return시 max() 한번 더 점검(!)

---