[BOJ] 1715 카드 정렬하기

2021. 7. 18. 17:22· Koala - 4기

우선순위 큐 문제를 별로 안풀어봐서인지, 혼자서는 솔직히 이 문제가 우선순위 큐인지 알아차리지 못했을 것 같습니다...

가장 중요한 것은 적어도 배열의 맨 앞의 값 두개는 항상 가장 작은 값이여야 할 것이라고 생각했고, 우선순위 큐를 이용해서 풀어주었습니다.

파이썬으로 작성한 코드

우선순위 큐에 집어 넣고 매번 pop을 통해 2개의 값을 꺼낸뒤 합을 total에 더하고 다시 큐에 넣어줬습니다. 그렇게 반복할 때 마다 큐의 길이가 1씩 줄어들게 되는데 queue의 길이가 1이면 반복문을 탈출시켜줄 수도 있지만 매번 비교를 해야되므로 if를 쓰거나 while에 조건을 다는 대신 try와 except로 빈 큐에 pop을 하려 했을 때 나는 에러를 이용해서 탈출시켜 주었습니다.

import sys
import heapq
input=sys.stdin.readline

n=int(input())

pq=[]
total=0

for i in range(n):
	heapq.heappush(pq,int(input()))

while True:
	try:
		n1=heapq.heappop(pq)
		n2=heapq.heappop(pq)
		total+=n1+n2
		heapq.heappush(pq,n1+n2)
	except: 
		break
	
print(total)

C++로 작성한 코드

#include<iostream>
#include<queue>
using namespace std;

int main()
{
    int n,num,n1,n2,len;
    int total=0;
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    priority_queue<int> pq;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>num;
        pq.push(-1*num);
    }
    
    while (1)
    {
        n1=pq.top();
        pq.pop();
        if (pq.empty()==1)
        {
            break;
        }
        n2=pq.top();
        pq.pop();
        total-=(n1+n2);
        pq.push(n1+n2);
    }

    cout<<total<<endl;
    
    return 0;
}

배열의 크기를 입력받아 그만큼 우선순위 큐에 넣어준 뒤 앞에서 숫자 2개를 빼고 그 합을 결과에 더해준뒤 다시 우선순위 큐에 푸쉬해주었습니다. 이것을 첫 번째 숫자를 뺐을 때 배열이 비어있을 때까지 반복시켜주고 그렇게 얻은 결과를 출력해주었습니다.

queue에 priority_queue가 있다는 것만 알고 만들었더니 큰수가 앞에 오도록 되어있어서 앞에 -를 붙여서 뒤집어주고 결과에 더할때 -를 붙여 활용했습니다. 파이썬의 pop()이랑 다르게 맨앞에 것을 가져가는 것을 몰라서 해맸네요..

저작자표시 (새창열림)

'Koala - 4기' 카테고리의 다른 글

[BOJ] 22116 창영이와 퇴근  (3) 2021.07.19
백준 22116 창영이와 퇴근 풀이  (0) 2021.07.19
[BOJ] 1715 카드 정렬하기  (0) 2021.07.18
[BOJ 1715번] : 카드 정렬하기  (0) 2021.07.18
[BOJ] 카드 정렬하기 1715번  (1) 2021.07.18
'Koala - 4기' 카테고리의 다른 글
  • [BOJ] 22116 창영이와 퇴근
  • 백준 22116 창영이와 퇴근 풀이
  • [BOJ] 1715 카드 정렬하기
  • [BOJ 1715번] : 카드 정렬하기
KauKoala
KauKoala
항공대 알고리즘 동아리 Koala 🥰
Koala항공대 알고리즘 동아리 Koala 🥰
KauKoala
Koala
KauKoala
전체
오늘
어제
  • 분류 전체보기 (1883)
    • 공지 게시판 (10)
    • 정보 게시판 (8)
    • Codeforce (15)
    • acm-icpc (6)
    • Koala - 1기 (16)
    • Koala - 2기 (111)
      • Programming Contest (1)
      • A반 (20)
      • B반 (39)
      • C반 (22)
      • 기초 강의 (18)
    • Koala - 3기 (10)
      • 기초 스터디 (7)
    • Koala - 4기 (67)
    • Koala - 5기 (144)
      • 기초 알고리즘 스터디 (75)
      • 코딩테스트 준비 스터디 (68)
    • Koala - 6기 (102)
      • 기초 알고리즘 스터디 (75)
      • 코딩테스트 준비 스터디 (25)
      • 모의 테스트 스터디 (1)
    • Koala - 7기 (167)
      • 기초 알고리즘 스터디 (97)
      • 코딩테스트 준비 스터디 (68)
      • 모의 테스트 스터디 (1)
    • Koala - 8기 (44)
      • 기초 알고리즘 스터디 (32)
      • 코딩테스트 준비 스터디 (10)
      • 코드포스 버츄얼 스터디 (0)
      • 프로그래머스 LV2 스터디 (0)
    • Koala - 9기 (205)
      • 기초 알고리즘 스터디 (138)
      • 코딩테스트 준비 스터디 (64)
      • 모의테스트 준비 스터디 (1)
    • Koala - 10기 (117)
      • 기초 알고리즘 스터디 (30)
      • 코딩테스트 준비 스터디 (86)
      • 모의테스트 준비 스터디 (1)
    • Koala - 11기 (151)
      • 기초 알고리즘 스터디 (46)
      • 코딩테스트 준비 스터디 (104)
      • 모의테스트 준비 스터디 (1)
    • Koala - 12기 (86)
      • 기초 알고리즘 스터디 (31)
      • 코딩테스트 준비 스터디 (55)
    • Koala - 13기 (119)
      • 기초 알고리즘 스터디 (52)
      • 코딩테스트 준비 스터디 (67)
    • Koala - 14기 (116)
      • 기초 알고리즘 스터디 (39)
      • 코딩테스트 준비 스터디 (77)
    • Koala - 15기 (138)
      • 기초 알고리즘 스터디 (73)
      • 코딩테스트 준비 스터디 (65)
    • Koala - 16기 (47)
      • 코딩테스트 기초 스터디 (16)
      • 코딩테스트 심화 스터디 (31)
    • Koala - 17기 (62)
      • 코딩테스트 기초 스터디 (15)
      • 코딩테스트 심화 스터디 (47)
    • Koala - 18기 (31)
      • 코딩테스트 기초 스터디 (11)
      • 코딩테스트 심화 스터디 (20)
    • Koala - 19기 (38)
      • 코딩테스트 기초 스터디 (7)
      • 코딩테스트 심화 스터디 (31)

블로그 메뉴

  • 홈
  • 태그
  • 방명록

공지사항

  • 🐨항공대 알고리즘 학회 Koala 3기 모집
  • 🐨항공대 알고리즘 학회 Koala 2기 모집
  • 소모임 소개

인기 글

태그

  • 백준
  • dp
  • 파이썬
  • dfs
  • C++
  • 백트래킹
  • BFS
  • BOJ

최근 댓글

최근 글

hELLO · Designed By 정상우.v4.3.0
KauKoala
[BOJ] 1715 카드 정렬하기
상단으로

티스토리툴바

단축키

내 블로그

내 블로그 - 관리자 홈 전환
Q
Q
새 글 쓰기
W
W

블로그 게시글

글 수정 (권한 있는 경우)
E
E
댓글 영역으로 이동
C
C

모든 영역

이 페이지의 URL 복사
S
S
맨 위로 이동
T
T
티스토리 홈 이동
H
H
단축키 안내
Shift + /
⇧ + /

* 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.