[백준] 10773번 : 제로

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

10773번: 제로

 

#include <stdio.h>
int arr[100000];
int idx = 0;
int* stack(int num)
{
	
	if (num == 0)
	{
		idx--;
		arr[idx] = 0;
	}
	else
	{
		arr[idx] = num;
		idx++;
	}
	return arr;
}
int main()
{
	int n, sum = 0;
	scanf("%d", &n);
	int num;
	for (int i = 0; i < n; i++)
	{
		scanf("%d", &num);
		int* arr = stack(num);
	}
	for (int i = 0; i < n; i++)
	{
		sum += arr[i];
	}
	printf("%d\n", sum);
}

> 어짜피 arr에 있는 모든 값의 합을 구하는게 답이기 때문에,

굳이 pop() 함수를 구현해서 return 을 하는 과정은 필요없다고 생각하여,

0이 입력되면 index값을 하나 줄여서, 지워야 하는 값을 0으로 갱신했다.

그렇게 되면 arr 배열의 모든 합인 sum을 구하는데에 있어서 영향을 끼치지 않게 된다.