본문 바로가기

prompt

[토요라운드] C. 사탕 (1812)

 

어떻게 풀지 고민하다가 이걸 다 더해보니 1 + 2 + 3 + 4 + 5를 구할 수 있었다. 여기서 기존에 있는 걸 빼면 되겠다 생각했음. 

 

n = 5일 때 고려해보니 위의 저 노란색만 빼면 되겠다 싶었다. 저 노란 덩어리는 [(i + 1 + N) % N], [(i - 2 + N) % N] 라서 이 인덱스의 수만 뺐는데 예시가 틀렸다ㅋㅋ 예시는 n의 크기가 3인데 여기서 두 덩어리를 빼면 값이 이상하게 나온다. 

 

 

보니깐 홀수 인덱스만 빼면 되겠다. 

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <bitset>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()

using namespace std;
using i64 = long long int;
using ii = pair<int, int>;
using ii64 = pair<i64, i64>;
using iii = tuple<int, int, int>;


int     main()
{
	int n;
	scanf("%d", &n);

	vector<int> v(n);
	for (int i = 0; i < n; i++)
		scanf("%d", &v[i]);

	i64 sum = 0;
	for (int i = 0; i < n; i++)
		sum += v[i];
	sum /= 2;

	for (int i = 0; i < n; i++)
	{
		i64 ans = sum;

		for (int j = 1; j < n; j += 2)
		{
			ans -= v[(i + j + n) % n];
		}

		printf("%lld\n", ans);
	}

}