백준 32642 당구 좀 치자 제발 (c++) : 피너클

https://www.acmicpc.net/problem/32642

구현문제다.

int n;
long long ans = 0;
long long angry = 0;

변수는 long long으로 설정해준다. 문제에서 권장해줬다.

cin >> n;

int b;
for (int i = 0; i < n; i++) {
	cin >> b;
	if (b == 1) angry += 1;
	else angry -= 1;

	ans += angry;
}

그리고 그냥 값 입력받으면서

1을 입력받으면 화가났으니 angry + 1을 해주고 아니면 화가 줄어들었으니 angry - 1을 해준다.

그후 ans에 angry를 더하면 된다.

#include <iostream>
#include <algorithm>

using namespace std;

int n;
long long ans = 0;
long long angry = 0;

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	cin >> n;

	int b;
	for (int i = 0; i < n; i++) {
		cin >> b;
		if (b == 1) angry += 1;
		else angry -= 1;

		ans += angry;
	}

	cout << ans << endl;

}

전체코드다.