4 5 1
1 2
1 3
1 4
2 4
3 4

1 2 4 3
1 2 3 4

5 5 3
5 4
5 2
1 2
3 4
3 1

3 1 2 5 4
3 1 4 2 5

1000 1 1000
999 1000

1000 999
1000 999

bfs는 큐로, dfs는 재귀(스택)으로 출력할 값들을 저장하면 된다.

#include <iostream>
#include <queue>
using namespace std;
#define MAX_VERTEX 1001
short graph[MAX_VERTEX][MAX_VERTEX];
char check_dfs[MAX_VERTEX];
char check_bfs[MAX_VERTEX];

int n ,m, v;
int from, to;

void dfs(int node){
    if(check_dfs[node]){
        return;
    } else{
        check_dfs[node] = 1;
        cout << node << " ";
        for(int i = 1; i <= n; i++){
            if(graph[node][i] && !check_dfs[i])
            dfs(i);
        }
    }

}

void bfs(int node){
    queue<short> q;
    int temp = 0;

    q.push(node);
    check_bfs[node] = 1;

    
    while(!q.empty()){

        temp = q.front();
        q.pop();
        cout << temp << " ";


        for(int i = 1; i <= n; i++){
            if(graph[temp][i] && !check_bfs[i]){
                q.push(i);
                check_bfs[i] = 1;
            }
        }

    }
}

int main(){
    
    cin >> n >> m >> v;

    for(int i = 0 ; i < m ; i++){
        cin >> from >> to;
        graph[from][to]=1;
        graph[to][from]=1;
    }
    dfs(v);
    cout << "\n";
    bfs(v);

    return 0;
}

vertex(정점)과 node는 같은 개념으로 사용하였다.

또한, bfs에서 같은 깊이에 있는 노드들을 작은 수부터 출력하라 하였는데,  애초에 다음 노드 탐색을 for문을 이용하여 1번 노드부터 n번 노드까지 탐색하고 push하였으므로 정렬할 필요는 없다.

6 5
1 2
2 5
5 1
3 4
4 6

2

6 8
1 2
2 5
5 1
3 4
4 6
5 4
2 4
2 3

1

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <stdlib.h>

#define MAX_VERTEX 1001

short graph[MAX_VERTEX][MAX_VERTEX]; // adjacency matrix
char check[MAX_VERTEX]; // check visitied node


void dfs(int i,int vertex_num) {
	
	check[i] = 1;
	for (int k = 1; k <= vertex_num; k++) {
		if(graph[i][k] && !check[k])
			dfs(i, k, vertex_num);
	}
	
}

int main() {
	int vertex_num, edge_num;
	int from, to;
	scanf("%d %d", &vertex_num, &edge_num);

	for (int i = 0; i < edge_num; i++) {
		scanf("%d %d", &from, &to);
		graph[from][to] = 1;
		graph[to][from] = 1;  //no direction
	}

	int count = 0;

	for (int i = 1; i <= vertex_num; i++) {
		
		if (!check[i]) {
			count++;
			dfs(i, vertex_num);
		}
		
	}

	printf("%d", count);

}

그래프 유형의 문제가 처음이라 꽤 애먹었는데, 

집합의 개수를 살릴때 그래프 기준이 아니라, 노드를 기준으로 count 해야하고, bfs 혹은 dfs는 연결되어있는 노드에 방문하기 위한 알고리즘이며 어떤 순서로 방문할껀지에 따라 알고리즘이 나뉜다.

2
10 8 17
0 0
1 0
1 1
4 2
4 3
4 5
2 4
3 4
7 4
8 4
9 4
7 5
8 5
9 5
7 6
8 6
9 6
10 10 1
5 5

5
1

1
5 3 6
0 2
1 2
2 2
3 2
4 2
4 0

2

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
int global_n, global_m;

void record_history_dfs(int** arr, int x, int y) {
	int new_x, new_y;
	arr[y][x] = 0;
	// x ,y+1
	new_x = x;
	new_y = y + 1;
	if (new_y < global_n && arr[new_y][new_x]) {
		record_history_dfs(arr, new_x, new_y);
	}

	// x ,y-1
	new_x = x;
	new_y = y - 1;
	if (new_y >= 0 && arr[new_y][new_x]) {
		record_history_dfs(arr, new_x, new_y);
	}
	// x+1 ,y
	new_x = x + 1;
	new_y = y;
	if (new_x < global_m && arr[new_y][new_x]) {
		record_history_dfs(arr, new_x, new_y);
	}
	// x-1 ,y
	new_x = x - 1;
	new_y = y;
	if (new_x >= 0 && arr[new_y][new_x]) {
		record_history_dfs(arr, new_x, new_y);
	}
}

	int count(int** arr, int m, int n) {
		int total_count = 0;
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++) {
				if (arr[i][j]) {
					record_history_dfs(arr, j, i);
					total_count++;
				}
			}
		}
		return total_count;
	}

	int main() {
		int size;
		int n, m, k;
		int x, y;
		int** arr = NULL;
		scanf("%d", &size);
		for (int i = 0; i < size; i++) {
			scanf("%d %d %d", &m, &n, &k);
			global_m = m;
			global_n = n;
			arr = (int**)malloc(sizeof(int*) * n);
			for (int j = 0; j < n; j++) {
				arr[j] = (int*)malloc(sizeof(int) * m);
			}
			for (int j = 0; j < n; j++) {
				for (int k = 0; k < m; k++) {
					arr[j][k] = 0;
				}
			}

			for (int j = 0; j < k; j++) {
				scanf("%d %d", &x, &y);
				arr[y][x] = 1;
			}

			printf("%d\n", count(arr, m, n));

		}
	}

재귀적으로 호출하는 dfs 알고리즘을 사용하였다.

우선 1을 만나면, 해당 좌표 기준 상, 하, 좌 , 우 좌표가 바운더리 내에 존재하는지 확인 후 해당 좌표도 1이라면, 모두 0으로 바꿔주는 작업을 한 후, count를 하는 방향으로 설계하였다.

int fibonacci(int n) {
    if (n == 0) {
        printf("0");
        return 0;
    } else if (n == 1) {
        printf("1");
        return 1;
    } else {
        return fibonacci(n‐1) + fibonacci(n‐2);
    }
}

3
0
1
3

1 0
0 1
1 2

2
6
22

5 8
10946 17711

해당 문제는 다이나믹 프로그래밍 알고리즘 문제로, 탑다운 혹은 바텀업 방식을 선택할 수 있다.

특징으론, 시간이 0.25초로 매우 빠듯하게 설정되어 있으므로, 효율적이로 구현해야 한다.

문제에서 n의 범위가 40까지밖에 되지 않으므로, 바텀업 방식으로 간단하게 구현하였다.

#include <iostream>
#include <stdlib.h>
using namespace std;
int main(){
    int one[41] = {0,};
    int zero[41] = {0,};

    zero[0] = 1;
    one[0] = 0;

    zero[1] = 0;
    one[1] = 1;

    zero[2] = 1;
    one[2] = 1;

    for(int i = 3; i <= 40; i++){
        zero[i] = zero[i-1] + zero[i-2];
        one[i] = one[i-1] + one[i-2];
    }

    int size = 0;
    cin >> size;
    int* user_input = (int*)malloc(sizeof(int)*size);
    for(int i = 0; i < size; i++)
        cin >> user_input[i];
    for(int i = 0; i < size; i++)
        cout << zero[user_input[i]] << " " << one[user_input[i]] << "\n";
}

인덱스가 2인 경우 또한 for문에 넣어도 되지만, 이해를 돕고자 위처럼 분리하여 선언하였다.

결국 n도 n-1, n-2로 분리되므로 1의 개수와 0의 개수는 수마다 고유하다고 생각하면 된다.

문제

팀 레드시프트는 대회 준비를 하다가 지루해져서 샌드박스 게임인 ‘마인크래프트’를 켰다. 마인크래프트는 1 × 1 × 1(세로, 가로, 높이) 크기의 블록들로 이루어진 3차원 세계에서 자유롭게 땅을 파거나 집을 지을 수 있는 게임이다.

목재를 충분히 모은 lvalue는 집을 짓기로 하였다. 하지만 고르지 않은 땅에는 집을 지을 수 없기 때문에 땅의 높이를 모두 동일하게 만드는 ‘땅 고르기’ 작업을 해야 한다.

lvalue는 세로 N, 가로 M 크기의 집터를 골랐다. 집터 맨 왼쪽 위의 좌표는 (0, 0)이다. 우리의 목적은 이 집터 내의 땅의 높이를 일정하게 바꾸는 것이다. 우리는 다음과 같은 두 종류의 작업을 할 수 있다.

1. 좌표 (i, j)의 가장 위에 있는 블록을 제거하여 인벤토리에 넣는다.
2. 인벤토리에서 블록 하나를 꺼내어 좌표 (i, j)의 가장 위에 있는 블록 위에 놓는다.

1번 작업은 2초가 걸리며, 2번 작업은 1초가 걸린다. 밤에는 무서운 몬스터들이 나오기 때문에 최대한 빨리 땅 고르기 작업을 마쳐야 한다. ‘땅 고르기’ 작업에 걸리는 최소 시간과 그 경우 땅의 높이를 출력하시오.

단, 집터 아래에 동굴 등 빈 공간은 존재하지 않으며, 집터 바깥에서 블록을 가져올 수 없다. 또한, 작업을 시작할 때 인벤토리에는 B개의 블록이 들어 있다. 땅의 높이는 256블록을 초과할 수 없으며, 음수가 될 수 없다.

입력

첫째 줄에 N, M, B가 주어진다. (1 ≤ M, N ≤ 500, 0 ≤ B ≤ 6.4 × 107)

둘째 줄부터 N개의 줄에 각각 M개의 정수로 땅의 높이가 주어진다. (i + 2)번째 줄의 (j + 1)번째 수는 좌표 (i, j)에서의 땅의 높이를 나타낸다. 땅의 높이는 256보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 땅을 고르는 데 걸리는 시간과 땅의 높이를 출력하시오. 답이 여러 개 있다면 그중에서 땅의 높이가 가장 높은 것을 출력하시오.

예제 입력 1 복사

3 4 99
0 0 0 0
0 0 0 0
0 0 0 1

예제 출력 1 복사

2 0

맨 오른쪽 아래의 블록을 제거하면 모두 높이가 0으로 고른 상태가 된다. 따라서 블럭을 한 번 제거하는 시간 2초가 소요된다.

예제 입력 2 복사

3 4 1
64 64 64 64
64 64 64 64
64 64 64 63

예제 출력 2 복사

1 64

인벤토리에 블록이 하나 있기 때문에, 맨 오른쪽 아래에 블록을 하나 채우면 된다.

예제 입력 3 복사

3 4 0
64 64 64 64
64 64 64 64
64 64 64 63

예제 출력 3 복사

22 63

인벤토리가 비어 있기 때문에, 맨 오른쪽 아래를 제외한 모든 좌표에서 블록을 하나씩 제거해야 한다.

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <stdlib.h>

typedef struct TimeHight {
	int time;
	int height;
}MyTimeHight;

//qsort compare함수, 1순위로 time값을 비교하고, time값이 같을 경우 2순위로 height 비교. time과 height가 같은 경우만 0 반환
int compare_time(const void* a, const void* b) {
	MyTimeHight* A = (MyTimeHight*)a;
	MyTimeHight* B = (MyTimeHight*)b;
	if (A->time > B->time)
		return 1;
	else if (A->time < B->time)
		return -1;
	else {
		if (A->height > B->height)
			return 1;
		else if (A->height < B->height)
			return -1;
		else {
			return 0;
		}
	}

}

//높이가 가장 높은 블럭의 높이 반환
int cal_highest(int** arr, int N, int M) { 
	int highest_num = 0;
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			if (arr[i][j] > highest_num)
				highest_num = arr[i][j];
		}
	}
	return highest_num;
}

//높이가 가장 낮은 블럭의 높이 반환
int cal_lowest(int** arr, int N, int M) {
	int lowest_num = 256;
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			if (arr[i][j] < lowest_num)
				lowest_num = arr[i][j];
		}
	}
	return lowest_num;
}

//집터를 goal 높이로 만드는데 걸리는 시간, 블럭이 부족하여 goal 높이로 만들 수 없을 경우 -1 리턴
int cal_time(int** arr, int N, int M, int B, int goal) {

	int time = 0; //리턴값인 총 소요 시간
	int need = 0; //쌓거나 빼야하는 블록 수
	int total_needed = 0; //총 소요 블럭 수
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			need = goal - arr[i][j];
			if (need < 0) { //블록을 빼야 하는 경우
				time += -2 * need; //음수 처리	
			}
			else if (need > 0) {
				time += need;
			}
			total_needed += need;
		}
	}
	if (total_needed <= B)
		return time;
	else
		return -1; //블록이 부족한 경우
}

//최소 시간 중 가장 높은 높이를 가지는 경우의 (시간, 높이)값을 리턴하는 함수. 블록이 부족한 경우는 높이와 시간을 int_max로 할당하도록 하였다.
//또한, 걸리는 시간을 저장하기 위한 anss의 malloc 동적 할당을 실패할 경우, 높이와 시간을 -1로 할당하도록 하였다.
MyTimeHight calculate(int** arr, int N, int M, int B) {

	int highest = cal_highest(arr, N, M);
	int lowest = cal_lowest(arr, N, M);
	int num_times = highest - lowest + 1;
	MyTimeHight* anss = (MyTimeHight*)malloc(sizeof(MyTimeHight) * num_times);
	MyTimeHight ans = { 0,0 };

	int goal = lowest;
	int result = 0;

	if (anss != NULL) {
		while (goal <= highest) {
			result = cal_time(arr, N, M, B, goal);
			if (result == -1) { //블록이 부족한 경우
				anss[goal - lowest].time = 2147483647;
				anss[goal - lowest].height = 2147483647;
			}
			else {
				anss[goal - lowest].time = result;
				anss[goal - lowest].height = goal;
			}
			goal++;
		}

		
		qsort(anss, num_times, sizeof(MyTimeHight), compare_time);
		if (num_times > 1) {
			if (anss[0].time == anss[1].time) {
				int i = 0;
				while (anss[i].time == anss[i + 1].time && i < num_times)
					i++;
				if (anss[i].time == anss[i + 1].time) {
					ans.time = anss[i + 1].time;
					ans.height = anss[i + 1].height;
				}
				ans.time = anss[i].time;
				ans.height = anss[i].height;
			}
			else {
				ans.time = anss[0].time;
				ans.height = anss[0].height;
			}
		}
		else {
			ans.time = anss[0].time;
			ans.height = anss[0].height;
		}
		free(anss);
		return ans;
	}
	else {
		ans.height = -1;
		ans.time = -1;
		return ans;
	}
}

int main() {
	int N, M, B;
	scanf("%d %d %d", &N, &M, &B);

	int** arr = (int**)malloc(sizeof(int*) * N);
	if (arr == NULL) {
		printf("malloc error\n");
		return -1;
	}

	for (int i = 0; i < N; i++)
		arr[i] = (int*)malloc(sizeof(int) * M);



	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			scanf("%d", &arr[i][j]);
		}
	}

	MyTimeHight ans = calculate(arr, N, M, B);

	if(ans.time >= 0 && ans.height >= 0) 
		printf("%d %d", ans.time, ans.height);
	else { //동적 할당 실패한 경우
		printf("Malloc Allocation failed\n");
		return -1;
	}

	for (int i = 0; i < N; i++)
		free(arr[i]);
	free(arr);

	return 0;


}

5
abcde

4739715

3
zzz

25818

1
i

9

University > 아주대학교 > 2018 Ajou Programming Contest > Division 2 C번

문제의 핵심은 변수 범위를 초과하지 않게 제곱 계산을 하는 것 인듯 하다.

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

const int M = 1234567891;
const int R = 31;

long square_mod(int times) {
	long temp = 1;
	for (int i = 0; i < times; i++) { //returns R^times mod (M)
		temp *= R;
		if (temp >= M) {
			temp %= M;
		}
	}
	return temp;
}

int hash(char* target, int size) {
	long temp = 0;
	//수로 바꾸기
	for (int i = 0; i < size; i++) {
		target[i] = (int)target[i] - 96;
	}
	for (int i = 0; i < size; i++) {
		temp += target[i] * square_mod(i);
		if (temp >= M) {
			temp %= M;
		}
	}
	return temp;
}

int main() {
	int size = 0;
	scanf("%d", &size);
	getchar();//개행 문자 제거

	char* target_string = (char*)malloc(sizeof(char) * size+1); 
	if (target_string == NULL)
		return -1;
	
	fgets(target_string, size + 1, stdin);
	target_string[strcspn(target_string, "\n")] = 0;

	printf("%d", hash(target_string, size));

	free(target_string);

	return 0;
}

5
0 4
1 2
1 -1
2 2
3 3

1 -1
1 2
2 2
3 3
0 4

해당 문제는 c의 stdlib헤더의 qsort함수를 사용하면 쉽게 풀 수 있다.

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <stdlib.h>

typedef struct Coordinate {
	int x;
	int y;
}MyCoordinate;

int compare(const void* a, const void* b) {
	MyCoordinate* A = (MyCoordinate*)a;
	MyCoordinate* B = (MyCoordinate*)b;
	if (A->y != B->y)
		return A->y - B->y;
	else
		return A->x - B->x;
}

int main() {
	int size;
	scanf("%d", &size);
	MyCoordinate* p = (MyCoordinate*)malloc(sizeof(MyCoordinate) * size);

	for (int i = 0; i < size; i++) {
		scanf("%d %d", &(p[i].x), &(p[i].y));
	}

	qsort(p, size, sizeof(MyCoordinate), compare);

	for (int i = 0; i < size; i++) {
		printf("%d %d\n", p[i].x, p[i].y);
	}


}

So when I die (the [first] I will see in (heaven) is a score list).
[ first in ] ( first out ).
Half Moon tonight (At least it is better than no Moon at all].
A rope may form )( a trail in a maze.
Help( I[m being held prisoner in a fortune cookie factory)].
([ (([( [ ] ) ( ) (( ))] )) ]).
 .
.

yes
yes
no
no
no
yes
yes

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

char check(char* start, char* end) {
	int check_small = 0;
	int check_big = 0;
	char* position_small = NULL;
	char* position_big = NULL;

	char* small_history[101] = { NULL };
	char* big_history[101] = { NULL };
	int small_history_index = 0;
	int big_history_index = 0;

	
	for (char* i = start; i <= end; i++) {
		if (*i == '(') {
			check_small++;
			position_small = i;
			small_history[small_history_index++] = position_small;
		}
		else if (*i == '[') {
			check_big++;
			position_big = i;
			big_history[big_history_index++] = position_big;
		}
		else if (*i == ')') {
			check_small--;
			if (check_small < 0 || check_big < 0) {
				return 0;
			}
			if (check(position_small + 1, i - 1) == 0)
				return 0;
			if (small_history_index >= 2) {
				small_history_index--;
				position_small = small_history[small_history_index-1];
			}
				
			else
				position_small = NULL;
			
		}
		else if (*i == ']') {
			check_big--;
			if (check_small < 0 || check_big < 0) {
				return 0;
			}
			if (check(position_big + 1, i - 1) == 0)
				return 0;
			if (big_history_index >= 2) {
				big_history_index--;
				position_big = big_history[big_history_index-1];
			}
				
			else
				position_big = NULL;
		}
		
	}
	if (check_small == 0 && check_big == 0) {
		return 1;
	}
	else {
		return 0;
	}
	
}

char check_gpt(char* start, char* end) {
	char stack[101]; // 스택으로 사용할 배열
	int stackIndex = 0; // 스택의 현재 인덱스
	for (char* i = start; i != end; i++) {
		if (*i == '(' || *i == '[') {
			// 여는 괄호를 스택에 추가
			stack[stackIndex++] = *i;
		}
		else if (*i == ')' || *i == ']') {
			// 스택이 비었거나, 짝이 맞지 않는 경우
			if (stackIndex == 0 || (stack[stackIndex - 1] == '(' && *i != ')') || (stack[stackIndex - 1] == '[' && *i != ']')) {
				return 0;
			}
			// 짝이 맞는 경우 스택에서 제거
			stackIndex--;
		}
	}
	// 스택이 비어있지 않다면 괄호의 짝이 맞지 않음
	return stackIndex == 0 ? 1 : 0;
}

int main() {
	char input[101] = { 0, };
	char* end_of_string = NULL;

	while (1) {
		fgets(input, sizeof(input), stdin);
		end_of_string = strrchr(input, '.');

		if (end_of_string == NULL) //'.'을 안친 경우
			continue;

		if (end_of_string == input) // '.'만 친 경우
			break;
		else {
			if (check_gpt(input, end_of_string))
				printf("yes\n");
			else
				printf("no\n");
		}
			
	}
	return 0;
}

본래 생각했던 풀이 방법은, '('와 '['의 위치를 계속 기록하고, 만약 괄호가 닫힌다면 괄호 사이에 있는 문자열을 인자로 재귀호출하여, 내부 문장이 적절한지 판단하는 방법이였다.

이전 괄호의 위치를 포인터 배열로 관리하며, 재귀호출로 인해 코드가 읽기 힘들고, 비효율적으로 동작하는 안좋은 코드가 만들어졌다.

반면 gpt가 작성한 코드는 stack을 이용해서 짝이 맞는 경우아 최종적으로 스택에 아무것도 남지 않을 경우 1을 리턴, 짝이 맞지 않는 경우 0을 리턴하는 코드를 작성하였다. 이는 훨씬 직관적이고 이해하기 쉬우며, 효율적이다.

함수를 직관적으로 작성하다 재귀함수를 사용해야 할 경우, 스택을 이용해볼 생각을 하는게 좋다고 한다.

하지만, 스택을 이용했을때 보기 더 힘들거나, 재귀 함수를 이용하므로써 tail recursion optimization을 받을 수 있는 경우엔 재귀 함수를 사용하는게 더 좋다고 한다.

자료구조를 배운 만큼, 사용해보려 노력해야겠다 ㅜ