[백준] 17128번 : 소가 정보섬에 올라온 이유 (C++)

17128번 : 소가 정보섬에 올라온 이유

---

 

문제)

소 N마리가 정보섬에 올라왔다!

소들은 정보섬 1층 앞마당에서 A1, A2, A3, ⋯, AN, A1의 순서대로 동그랗게 앉아 쉬고 있다. 각 소들에게는 품질 점수 Ai가 적힌 스티커가 붙어 있다. 욱제는 소 떼 옆에서 효빈이가 계산해 둔 어떤 계산 식을 발견했는데, 그것은 아래와 같다.

"S = ∑(Ai × Ai₊₁ × Ai₊₂ × Ai₊₃) (단, Ai₊₄ = Ai, Ai₊₅ = Ai₁, Ai₊₆ = Ai₂)"

풀어 쓰자면, 원형으로 둘러 앉은 소들에 대해서, 연속한 네 마리 소들의 품질 점수를 곱한 값을 모두 더한 것이다.

욱제는 효빈이가 학교를 떠나지 못하도록 심술부릴 작정이다. 욱제는 총 Q번에 걸쳐 어떤 i번째 소를 선택할 것이다. 그러고는 Ai가 적힌 스티커를 떼어내고, Ai × -1이 적힌 스티커를 붙일 작정이다. 그러면 효빈이는 Q번에 걸쳐서 S를 다시 계산해야 한다. 한 번 바꾼 스티커는 다음에 또 다시 바꾸지 않는 이상 계속 유지된다.

 

 

효빈이의 절친인 당신은 악동 욱제에게 괴롭힘 받는 효빈이를 도와 주기로 했다. 효빈이를 도와 S를 계산해 보자!

 

 

 

입력 :

첫째 줄에 소의 수를 나타내는 N과 욱제가 장난칠 횟수 Q가 주어진다.
둘째 줄에 N마리 소들의 품질 점수 Ai가 순서대로 주어진다.
셋째 줄에 욱제가 장난칠 Q개의 소의 번호 Qi가 순서대로 주어진다.

 

 

 

출력 :

 Q개의 줄에 걸쳐 다시 계산된 S의 값을 출력한다.

 

 

 

 

 

풀이)

 

// 17128. 소가 정보섬에 올라온 이유
#include <iostream>
#include <vector> 
 
using namespace std;
 
int n, q;
vector<int> vec;
vector<int> v;
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
 
    cin >> n >> q;
    vec.resize(n, 0);
    for (auto& ele : vec) cin >> ele;
 
    int origin = 0;
 
    for (int i = 0; i < n; i++)
    {
        int temp = 1;
        for (int j = i; j < i + 4; j++)
        {
            int t = j;
            if (t >= n) t %= n;
            temp *= vec[t];
        }
        v.push_back(temp);
        origin += temp;
    }
 
    int a;
    for (int i = 0; i < q; i++)
    {
        cin >> a;
        a--;
        int &ans = origin;
 
        for (int j = a - 3; j <= a; j++)
        {
            int t = j;
            if (t < 0) t = n + t;
            ans -= (v[t] * 2);
            v[t] *= -1;
        }
        cout << ans << '\n';
    }
 
    return 0;
}
 

---

출처 : https://www.acmicpc.net/problem/17128 

17128번: 소가 정보섬에 올라온 이유