(SWEA c++)2805. 농작물 수확하기

swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV7GLXqKAWYDFAXB&categoryId=AV7GLXqKAWYDFAXB&categoryType=CODE

SW Expert Academy

 

마름모 형태의 농작물을 더하면 된다.

for문을 이용해 두 부분으로 나눠서 값을 더해줬다.

 

우선 mid를 이용하여 배열의 중앙 부분을 계속해서 더한다.

그리고 중앙을 기준으로 왼쪽, 오른쪽 부분을 더해준다.

1 ~ mid 인 부분과 mid + 1 ~ n - 1 부분으로 나눠서 왼쪽, 오른쪽을 더하면 된다.

#include <iostream>
#include <vector>
#include <string>

using namespace std;

int main(void)
{
    int t;
    cin >> t;
    
    for(int i = 1; i <= t; i++)
    {
        int n, mid, ans = 0;
        string s;
        vector<string> v;
        
        cin >> n;
        mid = n / 2;
        
        for(int j = 0; j < n; j++)
        {
            cin >> s;
            v.push_back(s);
        }
        
        for(int j = 0; j < n; j++)
        {
            ans +=v[j][mid] - '0';
            
            if(j > 0 && j <= mid)
            {
            	for(int k = 1; k <= j; k++)
                {
                    ans += v[j][mid - k] - '0';
                    ans += v[j][mid + k] - '0';
                }   
            }
            else if(j > mid && j < n - 1)
            {
                for(int k = 1; k < n - j; k++)
                {
                    ans += v[j][mid - k] - '0';
                    ans += v[j][mid + k] - '0';
                }
            }
        }
        
        cout << "#" << i << " " << ans << endl;
    }
    return 0;
}