[SWEA 2805] 농작물 수확하기

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV7GLXqKAWYDFAXB 

SW Expert Academy

#include <cstdio>
#include <cstring>
int T;
int N;
int B[50][50];
char in[50][50];
int A[50][50];

void init()
{
	memset(B, 0, sizeof(B));
	memset(A, 0, sizeof(A));
	memset(in, '0', sizeof(in));
}


void input()
{
	scanf("%d", &N);
	for (int i = 0; i < N; i++) {
		scanf("%s", &in[i]);
		for (int j = 0; j < N; j++) {
			A[i][j] = in[i][j] - '0';
		}
	} 
}

int get_value()
{
	int total = 0;

	int sx = 0;
	int sy = N / 2;

	// 위에 삼각형 
	for (int n = 1; n <= N; n += 2) {
		int x = sx;
		int y = sy;
		for (int i = 1; i <= n; i++) {
			total += A[x][y];
			x = x;
			y += 1;
		}
		if (n == N) {
			sx += 1;
			sy += 1;
			break;
		}
		sx += 1;
		sy -= 1;
	}

	// 아래 삼각형 
	for (int n = N - 2; n >= 1; n -= 2) {
		int x = sx;
		int y = sy;
		for (int i = 1; i <= n; i++) {
			total += A[x][y];
			x = x;
			y += 1;
		}
		sx += 1;
		sy += 1;
	}
	return total;
}



int main()
{
	scanf("%d", &T);
	for (int t = 1; t <= T; t++) {
		init();
		input();
		printf("#%d %d\n", t, get_value());
	}
	return 0;
}