[Python] 함수 안에 함수 | 내부 함수 - 외부 함수

Python의 내부 함수를 이용한 풀이를 본적이 있어서

해당 방법으로 문제를 해결하고싶었다.

 

코테 문제 형식은 백준처럼 입력을 스스로 받는 형식이 아니라

프로그래머스처럼 함수내부를 작성하는 방식이기 때문이다

 

또한 DFS나 재귀함수를 사용할 때에

필요한 내용이라 생각이 들어서 이렇게 정리하게 되었다.

 

 

 

 

---

 

 

 

 

프로그래머스의 일반적인 형식을 먼저 살펴보자

해당 solution에 n인자를 넘겨주는 방식 채점이 진행될 것이다.

def solution(n):
	answer = n + 1
    return answer
    
print(solution(10))  # 11

 

만약 다음과 같은 상태에서 solution내부에 선언한 변수를

사용하고싶다면 어떻게 하겠는가?

 

함수 안에 함수를 작성할 수 있다!

다음은 node를 3개 방문하는 코드이다

def solution():
	
    def dfs(level):
    	if level != 3:
        	dfs(level+1)
            
    dfs(0)

여기서 solution함수는 외부함수

dfs함수는 내부함수라고 부른다.

 

당연히 solution도 함수이므로 호출해야 위의 코드를 실행시킬 수 있을 것이다

현 상태에서 solution을 호출해도 아무것도 출력하지 않기 때문에 따로 출력하지 않았다.

 

 

 

조금 더 나아가서

"dfs함수는 solution안에 있으니까 solution에 선언한 변수를 사용할 수 있지 않을까?"

하는 생각이 들었다면 박수👏

 

 

다음 코드를 보고 결과를 생각해보자

def solution():
    answer = 0:
    
    def dfs(level):
    	if level != 3:
        	answer += 1
        	dfs(level+1)
            
    dfs(0)
    return answer
    
    
print(solution())

정답은 3! 이라고 생각했다면 아쉽게도 틀렸다.

해당 코드에서는 dfs내부 answer += 1 부분에서 오류가 발생한다.

 

"이거 알려 주지 않은 변수라서 +1 못해"

라는 내용의 오류이다

 

solution내부에서 dfs불렀고, solution 아직 안 끝났는데?!할 수 있다.

 

 

 

 

 

코드를 하나만 더 보도록 하자

def solution():
    node = 'node'

    def dfs(level):
        if level != 3:
            print(node, level)
            dfs(level + 1)

    dfs(0)



solution()
#node 0
#node 1
#node 2

solution의 변수를 사용하여 각 노드를 출력하는

해당 코드는 문제없이 작동한다.

즉, 아까 생각한대로 내부함수에서는 외부함수의 변수를 사용할 수 있는 것이다

 

 

변수를 변경할 때, 함수는 지역 변수에서부터 넓혀가며

변수를 찾기 때문에 내부에 선언되지 않은 변수의 값을 변경하려하니 오류를 내는 것이다!

 

그래서 사용할 것은 nonlocal

def solution():
    answer = 0

    def dfs(level):
        nonlocal answer
        if level != 3:
            answer += 1
            dfs(level + 1)

    dfs(0)
    print(answer)

solution()
# 3

이제서야  3개의 node를 거쳤다는 의미의 알맞은 답 3이 나왔다..!

 

 

nonlocal <변수명>은

지역변수가 아닌 밖에 선언된 변수를 사용하겠다는 의미이다

 

 

 

 

 

지금까지 학습한 내용을 응용한 다양한 문제를 풀게 될 것이 기대된다..!

 

---

REFERENCE

https://dojang.io/mod/page/view.php?id=2365 

파이썬 코딩 도장: 33.2 함수 안에서 함수 만들기