득이공간
[백준 C++] 2357 최솟값과 최댓값 - 세그먼트트리 본문
#include <iostream>
#include <cmath>
using namespace std;
int Count, StartIndex;
int MinTree[131072 * 2];
int MaxTree[131072 * 2];
void SearchTree(int N, int Start, int End)
{
int CurStart = StartIndex + Start;
int CurEnd = StartIndex + End;
int Min = 1000000001;
int Max = 0;
while (CurStart <= CurEnd)
{
if (CurStart % 2 == 1)
{
Min = min(Min, MinTree[CurStart]);
Max = max(Max, MaxTree[CurStart]);
}
if (CurEnd % 2 == 0)
{
Min = min(Min, MinTree[CurEnd]);
Max = max(Max, MaxTree[CurEnd]);
}
CurStart = (CurStart + 1) / 2;
CurEnd = (CurEnd - 1) / 2;
}
cout << Min << ' ' << Max << '\n';
}
void InitTree(int N)
{
while (pow(2, Count) < N)
{
++Count;
}
StartIndex = pow(2, Count);
for (int i = StartIndex; i < StartIndex + N; ++i)
{
int Number;
cin >> Number;
MinTree[i] = Number;
MaxTree[i] = Number;
}
for (int i = StartIndex - 1; i > 0; --i)
{
if (MinTree[2 * i] == 0)
{
MinTree[i] = MinTree[2 * i + 1];
}
else if (MinTree[2 * i + 1] == 0)
{
MinTree[i] = MinTree[2 * i];
}
else
{
MinTree[i] = min(MinTree[2 * i], MinTree[2 * i + 1]);
}
MaxTree[i] = max(MaxTree[2 * i], MaxTree[2 * i + 1]);
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(NULL); cout.tie(NULL);
int N, M;
cin >> N >> M;
InitTree(N);
for (int i = 0; i < M; ++i)
{
int Start, End;
cin >> Start >> End;
SearchTree(N, Start - 1, End - 1);
}
}
세그먼트 트리를 이용해서 정해진 구간의 최소, 최대값을 구하는 문제입니다.
최소값 트리와 최대값 트리를 만들어서 각각 값을 따로 저장하도록 풀었습니다.
'PS > 알고리즘 문제풀이' 카테고리의 다른 글
[백준 C++] 11505 구간 곱 구하기 - 세그먼트트리 (1) | 2024.02.12 |
---|---|
[백준 C++] 10868 최솟값 - 세그먼트트리 (1) | 2024.02.12 |
[백준 C++] 2042 구간 합 구하기 - 세그먼트트리 (0) | 2024.02.11 |
[백준 C++] 1991 트리 순회 - 이진트리 (0) | 2024.02.10 |
[백준 C++] 1946 신입 사원 - 정렬 (0) | 2024.02.09 |