백준(baekjoon) [C] - 10773번: 제로

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

10773번: 제로

---

#include <stdio.h>  // 제로
#include <stdlib.h>

typedef struct
{
	int* stk;
	int ptr;
}STACK;

int is_empty(STACK* s)
{
	if (s->ptr <= 0)
		return 1;
	else
		return 0;
}

void push(STACK* s, int num, int k, int *sum)
{
	if (s->ptr <= k)
	{
		s->stk[s->ptr++] = num;
		*sum += num;
	}
}

void pop(STACK* s, int *sum)
{
	if (!is_empty(s))
	{
		*sum -= s->stk[--s->ptr];
	}
}


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

	STACK s;
	s.stk = malloc(k * sizeof(int));
	s.ptr = 0;

	int num, sum = 0;
	for (int i = 0; i < k; i++)
	{
		scanf("%d", &num);
		if (num)
			push(&s, num, k, &sum);
		else
			pop(&s, &sum);
	}

	printf("%d", sum);

	return 0;
}

---

MEMO

-- 문제를 읽고 바로 스택이 떠올랐다

-- 스택의 구현으로 간단하게 해결가능

 

!! 기본 다지기 !!