GDSC Algorithm Study Week 4

Author: jhjalison01

04-240409/jhjalison01/2143.py

@@ -0,0 +1,34 @@
+#백준 2143번 두 배열의 합
+
+import sys
+input=sys.stdin.readline
+import bisect
+
+T = int(input())
+n=int(input())
+a=list(map(int,input().split()))
+m=int(input())
+b=list(map(int,input().split()))
+
+#배열 a,b의 부 배열의 합 모든 경우 구하기
+a_sum,b_sum=[],[]
+for i in range(n):
+  for j in range(i+1,n+1):
+    a_sum.append(sum(a[i:j]))
+    
+for i in range(m):
+  for j in range(i+1,m+1):
+    b_sum.append(sum(b[i:j]))
+
+#이분 탐색을 위해 정렬
+a_sum.sort()
+b_sum.sort()
+
+answer=0
+for i in range(len(a_sum)):
+  tmp=T-a_sum[i]
+  left = bisect.bisect_left(b_sum, tmp)
+  right = bisect.bisect_right(b_sum, tmp)
+  answer+=right-left #경우의 수 추가
+
+print(answer)

04-240409/jhjalison01/A.py

@@ -0,0 +1,16 @@
+#백준 14697번 방 배정하기
+
+import sys
+input=sys.stdin.readline
+
+a,b,c,n=map(int,input().split())
+result=0
+
+for i in range(n//a+1):
+  for j in range(n//b+1):
+    for k in range(n//c+1):
+      #합이 n이 될 경우
+      if i*a+j*b+k*c==n:
+        result=1
+        
+print(result)

04-240409/jhjalison01/B.py

@@ -0,0 +1,40 @@
+#백준 10597번 방 배정하기
+
+import sys
+input=sys.stdin.readline
+
+nums=input().rstrip()
+#n: 숫자 개수
+if len(nums)<10:
+  n=len(nums)
+else:
+  n=9+(len(nums)-9)//2
+
+visited=[False for _ in range(n+1)]
+
+def dfs(idx,arr):
+  #다 구한 경우 복구된 수열 출력
+  if idx==len(nums):
+    print(*arr)
+    exit()
+
+  #한 자릿수 체크
+  num1 = int(nums[idx])
+  if not visited[num1]:
+    visited[num1]=True
+    arr.append(num1)
+    dfs(idx+1,arr)
+    visited[num1] = False
+    arr.pop()
+    
+  #두 자릿수 체크
+  if idx+1<len(nums):
+    num2 = int(nums[idx:idx+2])
+    if num2<=n and not visited[num2]:
+      visited[num2]=True
+      arr.append(num2)
+      dfs(idx+2,arr)
+      visited[num2]=False
+      arr.pop()
+
+dfs(0,[])

04-240409/jhjalison01/C.py

@@ -0,0 +1,24 @@
+#백준 14675번 단절점과 단절선
+
+import sys
+input=sys.stdin.readline
+
+n=int(input())
+
+graph=[[] for _ in range(n+1)]
+for _ in range(n-1):
+  a,b=map(int,input().split())
+  graph[a].append(b)
+  graph[b].append(a)
+
+q=int(input())
+for _ in range(q):
+  t,k=map(int,input().split())
+  if t==1: #k번 정점이 단절점인지
+    if len(graph[k])<2:
+      print("no")
+    else:
+      print("yes")
+  else: #k번 정점이 단절선인지
+    #모든 간선은 노드 2개를 연결하고 있기 때문에 항상 yes
+    print("yes")

04-240409/jhjalison01/D.py

@@ -0,0 +1,26 @@
+#백준 3273번 두 수의 합
+
+#n 입력받기
+n=int(input())
+#수열 입력받기
+data=list(map(int,input().split()))
+#x 입력받기
+x=int(input())
+data.sort() #수열 오름차순으로 정렬
+answer=0
+
+front=0 #앞쪽 포인터
+back=n-1 #뒤쪽 포인터
+
+while front<back:
+  #두 수를 더했을 때 x가 나온 경우
+  if data[front]+data[back]==x:
+    answer+=1
+  #두 수를 더했을 때 x보다 작은 경우
+  elif data[front]+data[back]<x:
+    front+=1 #포인터 뒤로 이동
+    continue
+  #포인터 앞으로 이동
+  back-=1
+    
+print(answer)

04-240409/jhjalison01/E.py

@@ -0,0 +1,24 @@
+#백준 21318번 피아노 체조
+
+import sys
+input=sys.stdin.readline
+
+#악보의 개수
+n=int(input())
+#악보 난이도
+diff = list(map(int,input().split()))
+
+dp=[0]*n
+for i in range(1,n):
+  if diff[i-1]>diff[i]:
+    dp[i]=dp[i-1]+1
+  else:
+    dp[i]=dp[i-1]
+
+#질문 개수
+q=int(input())
+
+for _ in range(q):
+  x,y=map(int,input().split())
+  #x부터 y까지 틀린 악보 개수 출력
+  print(dp[y-1]-dp[x-1])

04-240409/jhjalison01/F.py

@@ -0,0 +1,24 @@
+#백준 1757번 달려달려
+
+import sys
+input=sys.stdin.readline
+
+n,m=map(int,input().split())
+dis=[0]
+for _ in range(n):
+  dis.append(int(input()))
+
+#dp[i][j] - 달린 거리
+#: i - i분, j - 지침 지수
+dp=[[0]*(m+1) for _ in range(n+1)]
+
+for i in range(1,n+1):
+  #직전에도 쉬고 현재 분에도 쉰 거리
+  dp[i][0] = dp[i-1][0]
+  for j in range(1,m+1):
+    #i분까지 달린 거리
+    dp[i][j] = dp[i-1][j-1]+dis[i]
+    dp[i][0] = max(dp[i][0],dp[i-j][j])
+
+#최대 거리 출력
+print(dp[-1][0])

04-240409/jhjalison01/G.py

@@ -0,0 +1,37 @@
+#백준 2606번 바이러스
+
+import sys
+
+n=int(sys.stdin.readline())
+m=int(sys.stdin.readline())
+
+graph=[]
+
+#그래프를 리스트 자료형으로 표현
+for i in range(n+1):
+  graph.append([])
+
+#각 노드가 연결된 정보를 표현
+for i in range(m):
+  a,b=map(int,sys.stdin.readline().split())
+  graph[a].append(b)
+  graph[b].append(a)
+
+#각 노드가 방문된 정보를 리스트 자료형으로 표현
+visited=[False]*(n+1)
+
+#1번 컴퓨터는 제외해야 하므로 초기값을 -1로 설정한다.
+result=-1
+
+#dfs 메소드 정의
+def dfs(graph,v,visited):
+  visited[v]=True
+  global result
+  result+=1
+  for i in graph[v]:
+    if not visited[i]:
+      dfs(graph,i,visited)
+  
+
+dfs(graph,1,visited)
+print(result)

04-240409/jhjalison01/H.py

@@ -0,0 +1,36 @@
+#백준 12761번 돌다리
+
+import sys
+input=sys.stdin.readline
+from collections import deque
+#스카이 콩콩의 힘, 동규의 현재위치, 주미의 현재위치
+a,b,n,m=map(int,input().split())
+        
+graph=[0 for _ in range(100001)]
+visited=[False for _ in range(100001)]
+#-1,+1,+a,-a,+b,-b, a배, b배
+dx=[-1,1,a,-a,b,-b,a,b]
+
+def bfs(a):
+  q=deque([a])
+  visited[a]=True
+
+  while q:
+    x=q.popleft()
+
+    for i in range(8):
+      #6가지 경우는 덧셈 연산
+      if i<6:
+        nx=x+dx[i]
+      else:
+        #나머지는 곱셈 연산
+        nx=x*dx[i]
+      #범위 내에 있고 탐색하지 않은 경우
+      if 0<=nx<=100000 and not visited[nx]:
+        q.append(nx)
+        visited[nx]=True
+        graph[nx]=graph[x]+1
+        
+        
+bfs(n)
+print(graph[m])

04-240409/jhjalison01/I.py

@@ -0,0 +1,46 @@
+#백준 14923번 미로 탈출
+
+import sys
+input=sys.stdin.readline
+from collections import deque
+
+dx=[-1,1,0,0]
+dy=[0,0,-1,1]
+
+n,m=map(int,input().split())
+
+hx,hy=map(int,input().split())
+ex,ey=map(int,input().split())
+
+graph=[]
+for _ in range(n):
+  graph.append(list(map(int,input().split())))
+
+#visited[x][y][magic]: (x,y) 지점에 남은 마법 횟수가 magic 일 경우 도달한 적이 있는지
+visited=[[[False]*2 for _ in range(m)] for _ in range(n)]
+visited[hx-1][hy-1][1]=True
+
+def bfs():
+  q=deque()
+  q.append((hx-1,hy-1,1,0))
+  
+  while q:
+    #magic: 남은 마법 횟수, cost: 걸린 시간
+    x,y,magic,cost=q.popleft()
+    if x==ex-1 and y==ey-1:
+      return cost
+
+    for i in range(4):
+      nx,ny=x+dx[i],y+dy[i]
+      if 0<=nx<n and 0<=ny<m and not visited[nx][ny][magic]:
+        #(nx,ny)가 벽이고 마법을 쓸 수 있을 경우 - 마법 사용
+        if magic==1 and graph[nx][ny]==1:
+          visited[nx][ny][0]=True
+          q.append((nx,ny,0,cost+1))
+        #(nx,ny)가 빈 칸일 경우 - 그냥 이동
+        elif graph[nx][ny]==0:
+          visited[nx][ny][magic]=True
+          q.append((nx,ny,magic,cost+1))
+  return -1
+
+print(bfs())