[백준 10773] 제로

[문제]

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

10773번: 제로

[코드]

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <stdlib.h> // malloc과 짝꿍
#include <string.h>
int main(void)
{
	long long n = 0, sum = 0, k = 0, remove = 0; // 개수, 합계, 입력받을 장소, save에 저장할 순번

	scanf("%lld", &n);

	long long* save = (long long*)malloc(sizeof(long long) * n); // 이전것들 저장

	for (int i = 0; i < n; i++) {
		scanf("%lld", &k);
		
		if (k == 0) {
			sum -= save[remove - 1];
			remove--;
		}
		else {
			sum += k;
			save[remove++] = k;
		}
	}

	printf("%lld", sum);
}