问题抽取
在做LeetCode 563题,二叉树的坡度,遇到了一个python的语法问题:
如何自定义函数的时候不返回值,但仍能更改参数
这个问题在C/C++里面的常规做法是引用传递,引用传递是指在调用函数时将实际参数的地址传递到函数中,在函数中直接更改该地址上的值,也就是一定更改了实际参数。但是在python里面还不知道怎么解决
出错过程
自己最开始写的代码如下:
class Solution(object):
def findTilt(self, root):
res = 0
def helper(root, res):
if root == None:
return 0
sum_left = helper(root.left, res)
sum_right = helper(root.right, res)
res += abs(sum_right - sum_left)
return root.val + sum_left + sum_right
helper(root, res)
return res
出错原因
查资料发现,函数运行过程中实际上是为形参开辟了一块新的内存空间,和实际参数已经没有关系,所以不会更改实际参数,在函数运行结束后形参也会消亡
通过如下代码测试可以看到,实参和形参是两块内存空间:
a = 1
print(id(a))
def util(a):
print(id(a))
a += 1
util(a)
print('a=', a)
执行结果:
140717689577872
140717689577904
a = 1
解决方案
后来看了官方题解,发现其解决方法是——使用构造函数和实例属性,把res塞进构造函数里面作为实例属性,然后再在成员函数里面更改实例属性self.res,代码如下:
class Solution:
def __init__(self):
self.res = 0
def findTilt(self, root):
self.dfs(root)
return self.res
def dfs(self, node):
if not node:
return 0
sum_left = self.dfs(node.left)
sum_right = self.dfs(node.right)
self.res += abs(sum_left - sum_right)
return sum_left + sum_right + node.val
同时又联想到了实例属性和类属性的区别,于是查了资料复习,如下: https://blog.csdn.net/qq_41867145/article/details/107304112
后续学习
在看官方题解的其他语言描述答案时,发现其他语言并没有这个问题,但是限于现在基本编程语言和语法的不熟悉,不知道其具体的实现形式,猜测应该都是通过面向对象编程,更改并返回实例属性的方式完成的。
后续在学习其他语言时进一步学习回顾。
|