pythonの特定の変数のスコープがおかしい

二分探索木を作りたい

pythonで二分探索木について勉強しているとき、二分探索木を作る関数を作った際に特定の変数だけ使えません。

発生している問題・エラーメッセージ

    #ファイル名
    addBST(1)
    #ファイル名
    print(newIdx)
UnboundLocalError: local variable 'newIdx' referenced before assignment

該当のソースコード

Python3
1class BST:
2    def __init__(self):
3        data = 0
4        left = 0
5        right = 0
6
7tree = [0]*10
8for i in range(len(tree)):
9    tree[i] = BST()
10rootIdx = 0
11newIdx = 0
12
13def addBST(data):
14    currentIdx = 0
15    addFlag = False
16    print(newIdx)
17    tree[newIdx].data = data
18    tree[newIdx].left = -1
19    tree[newIdx].right = -1
20    if newIdx != rootIdx:
21        currentIdx = rootIdx
22        addFlag = False
23        while True:
24            if data < tree[currentIdx].data:
25                if tree[currentIdx].left == -1:
26                    tree[currentIdx].left = newIdxaddFlag = True
27                else:
28                    currentIdx = tree[currentIdx].left
29            else:
30                if tree[currentIdx].right == -1:
31                    tree[currentIdx].right = newIdx
32                    addFlag = True
33                else:
34                    currentIdx = tree[currentIdx].right
35            if addFlag == False:
36                break
37    newIdx += 1
38
39addBST(1)
40addBST(3)
41addBST(2)

試したこと

print関数の引数を他の変数(rootIdxや新しい変数tmp)などに変えたところ、正常に出力されました。

補足情報

環境
mojave 10.14.4
python 3.5.4

行動規範の内容に同意します

回答2件

ベストアンサー

関数の中である変数名に対して代入文を使っていれば、その変数はローカル変数になります。
(ただの代入文の話。累積代入文や添字表記への代入は別途議論が必要)

ローカル変数だということは、ローカルスコープ内で名前解決しないといけないということです。つまり、関数ブロックの中で代入より先に参照すればエラーになります。

同名のグローバル変数がたまたまあったとしても、それはローカル変数の名前解決には関係のないことです。この考えかたは理解してください。
(ただのNameErrorにならないのがむしろ優しいといったところ)

それで、やりたいことが「この変数はグローバル変数として扱いたい」なら関数ブロックの先頭にglobal宣言を追加してください。ローカル変数でいいなら、参照の前に何らかの値を入れれば可です。

投稿2019/10/27 21:54

hayataka2049

総合スコア30933

インデントが、おかしくなっているのか意図的にしているのかわかりませんが、
以下のようにすれば、エラーは解消することができます。

ただ、後々newIdxという変数を使うことがあるのであれば、設計を見直すことを提案します。

python
1class BST:
2    def __init__(self):
3        data = 0
4        left = 0
5        right = 0
6
7tree = [0]*10
8for i in range(len(tree)):
9    tree[i] = BST()
10
11
12def addBST(data):
13    rootIdx = 0
14    newIdx = 0
15    currentIdx = 0
16    addFlag = False
17    print(newIdx)
18    tree[newIdx].data = data
19    tree[newIdx].left = -1
20    tree[newIdx].right = -1
21    if newIdx != rootIdx:
22        currentIdx = rootIdx
23        addFlag = False
24        while True:
25            if data < tree[currentIdx].data:
26                if tree[currentIdx].left == -1:
27                    tree[currentIdx].left = newIdxaddFlag = True
28                else:
29                    currentIdx = tree[currentIdx].left
30            else:
31                if tree[currentIdx].right == -1:
32                    tree[currentIdx].right = newIdx
33                    addFlag = True
34                else:
35                    currentIdx = tree[currentIdx].right
36            if addFlag == False:
37                break
38    newIdx += 1
39
40addBST(1)
41addBST(3)
42addBST(2)