3.1. 최대 힙 ¶
#include <stdio.h>
int Heap[200000];
int idx=1;
void push(int x);
int pop();
void balance(int now_idx);
int main() {
int N, i, O;
scanf("%d", &N);
for(i=0;i<N;i++){
scanf("%d", &O);
if(O==0){
printf("%d\n", pop());
}
else{
push(O);
}
}
return 0;
}
void push(int x){
Heap[idx] = x;
balance(idx);
idx++;
}
int pop(){
if(idx==1) return 0;
int out = Heap[1];
Heap[1] = Heap[idx-1];
Heap[idx-1] = 0;
balance_down(1);
idx--;
return out;
}
void balance(int now_idx){
int tmp;
if(now_idx==1) return;
if(Heap[now_idx]>Heap[now_idx/2]){
tmp = Heap[now_idx];
Heap[now_idx] = Heap[now_idx/2];
Heap[now_idx/2] = tmp;
}
balance(now_idx/2);
}
void balance_down(int now_idx){
int tmp, max_idx;
if(now_idx>=idx) return;
max_idx = Heap[now_idx*2]>Heap[now_idx*2+1] ? now_idx*2 : now_idx*2 +1;
if(Heap[now_idx]<Heap[max_idx]){
tmp = Heap[now_idx];
Heap[now_idx] = Heap[max_idx];
Heap[max_idx] = tmp;
balance_down(max_idx);
}
}
3.2. DFS와 BFS ¶
- 큐 구현은 안하고 C++ STL 쓸께요.
#include <stdio.h>
#include <queue>
using namespace std;
int N, M, V;
int Ver[1001][1001]={0};
int Ver1[1001][1001]={0};
int Already[1001];
queue<int> q;
void DFS(int Node);
void BFS();
int main() {
int A, B, i;
scanf("%d %d %d", &N, &M, &V);
for(i=0;i<M;i++){
scanf("%d %d", &A, &B);
Ver[A][B]=1;
Ver[B][A]=1;
Ver1[A][B]=1;
Ver1[B][A]=1;
}
DFS(V);
printf("\n");
for(i=1;i<=N;i++){
Already[i]=0;
}
q.push(V);
while(!q.empty()) {
BFS();
}
return 0;
}
void DFS(int Node){
int i;
if(Already[Node]) return;
printf("%d ", Node);
Already[Node]=1;
for(i=1;i<=N;i++){
if(Ver[Node][i]){
Ver[Node][i]=0;
Ver[i][Node]=0;
DFS(i);
}
}
}
void BFS(){
int i;
int Node = q.front();
if(Already[Node]) {
q.pop();
return;
}
q.pop();
printf("%d ", Node);
Already[Node]=1;
for(i=1;i<=N;i++){
if(Ver1[Node][i]){
Ver1[Node][i]=0;
Ver1[i][Node]=0;
q.push(i);
}
}
}










