프로그래머스 - flag에 따라 다른 값 반환하기

---

package main

import (
	"fmt"
)

func main() {
	a := -4
	b := 7
	flag := true
	fmt.Print(solution(a, b, flag))
}

func solution(a int, b int, flag bool) int {

	if flag == true {
		return a + b
	} else {
		return a - b
	}

}