본문 바로가기

분류 전체보기101

[segment tree] 1849번 순열 https://www.acmicpc.net/problem/1849 자료구조 Segment Tree 사용 언어 Python 시간 복잡도 $O(10^5log(10^5))$ 풀이 포인트 BOJ 2243번 풀이와 거의 동일 두 가지 다른 점은 update 함수를 따로 구현하면 TIL이 발생하여 query 함수에서 이를 처리해줘야 한다는 점과 query 함수에서 분기하는 if 조건이 =을 포함하지 않는다는 점 단순 for문 풀이: TIL BOJ 2243번 풀이와 마찬가지로 왜 segment tree 자료구조를 사용해야하는지에 대해서 생각해보자. for문을 이용해볼 때, 시간 복잡도가 어떻게 나오는지 살펴본다. 입력에는 수열 A가 주어지고 A[i]는 그 수열에서 i 앞에 있는 수(i 왼쪽에 있는 수 들 중,)들 중.. 2022. 4. 9.
[segment tree] 2243번 사탕상자 https://www.acmicpc.net/problem/2243 자료구조 Segment Tree 사용 언어 Python 시간 복잡도 $O(10^5 log(10^6))$ 풀이 포인트 트리의 i번째 노드 값이 1~i 사탕 맛의 개수의 합을 의미하도록 트리 설정 왼쪽 자식 노드의 값과 오른쪽 자식 노드의 값을 k와 비교하면서 왼쪽 자식 노드와 오른쪽 자식 노드 중 하나를 선택함 Segment Tree 이 문제에서 segment tree 자료구조를 사용해야하는 이유에 대해서 생각해보자. 굳이 자료구조를 사용해서 어렵게 풀지 않아도 될 상황이면, 머리 아프게 자료구조를 생각하지 않아도 되기 때문이다. 자료구조를 사용하지 않고 평범한 for문을 돌렸을 때, 시간 복잡도가 어떻게 나오는지 생각해보자. 수정이가 사탕.. 2022. 4. 8.
[segment tree] 11505번 구간 곱 구하기 https://www.acmicpc.net/problem/11505 자료구조 segment tree 풀이 포인트 segment tree 자료구조를 이용해서 $O(logn)$ 시간 복잡도로 문제 풀기 (segment tree 자료구조 설명) modular 연산에 유의하여 update 함수 변경하기 segment tree update 함수 기존의 누적 합 문제에서는 새로 바꿀 수와 기존의 수와의 차이 (diff)를 구하고, 해당 인덱스가 포함되는 노드에 diff를 더해주는 식으로 update 함수가 작동했다. 하지만 이 문제에서는 누적 곱을 구하는 문제이고 중간에 modular 연산이 들어가기 때문에 이런 방식으로 할 수가 없다. 만약에 누적합 문제와 유사하게 누적곱의 update 함수를 구성하면 아래와 같.. 2022. 3. 20.
Segment Tree Motivation segment tree 자료구조는 아래의 문제를 해결하는데 사용된다. 1. 1차원 배열의 i번째 수부터 j번째 수까지의 연속 합을 구해야할 때 (총 $m$번) 2. 배열의 i번째 수가 변경되는 쿼리를 같이 처리해야할 때 (총 $k$번) 1번의 문제만 해결해야한다면, 부분 합 (Prefix Sum) 문제로 치환하여, $O(n+m)$의 시간 복잡도로 해결할 수 있다. 하지만, 2번의 문제까지 추가가 된다면 상황이 복잡해진다. 부분 합을 이미 구해 놓은 상태에서, i번째 수가 변경된다면 부분 합을 다시 업데이트해야한다. 즉, $k$번만큼 배열의 수가 변경될때마다 부분 합을 업데이트해야하므로 여기서 $O(nk)$의 시간 복잡도가 발생한다. 여기에 m개의 연속 합을 구해야하므로 최종 시간 복잡.. 2022. 3. 19.
[구현] 2174번 로봇 시뮬레이션 https://www.acmicpc.net/problem/2174 알고리즘 구현, 시뮬레이션 풀이 포인트 파이썬 class를 이용해서 구현하기 구현, 시뮬레이션 로봇의 위치, 방향, 번호의 정보를 가지는 robot class를 만든다. $B \times A$ 크기의 land 2차원 배열을 선언하고, $(x,y)$ 좌표에 로봇이 있을 경우, 해당 위치에 로봇 클래스를 추가한다. 로봇의 개수만큼의 크기를 가지는 robot_info 1차원 배열을 선언하고 i번째 위치에 i번 로봇 클래스를 추가한다. 문제의 명령을 그대로 수행한다. 벽에 만나거나, 다른 로봇을 만나면 경고문과 함께 프로그램을 종료하고 그렇지 않다면 land 배열에서 로봇을 옮기거나 로봇 클래스의 방향을 수정해준다. [Source Code] a,.. 2022. 3. 17.
[Greedy] 12940번 A와 B https://www.acmicpc.net/problem/12904 알고리즘 Greedy 풀이 포인트 문자열 S를 T로 만들기 위해서는 다양한 경로를 따져야 하지만, T에서 S를 만드는 경로는 하나라는 점. Greedy 이 문제를 처음 접했을 때, 가장 먼저 떠오른 풀이는 bfs이었다. 즉, 현재 상태에서 두 가지 선택권이 있고, 각 상태를 만들기 위한 최소 연산 수를 기록하며 bfs를 한다면, 문자열 T를 만들 수 있는지 여부를 구할 수 있다고 생각했다. 전형적인 bfs 문제라고 생각하고 아래와 같이 코드를 짰다. [Source Code] from collections import deque def func_1(x): return x+'A' def func_2(x): return x[::-1]+'B' .. 2022. 3. 17.