如何递归检查对象中的属性？

Question

我一直在做电路模拟器，有点像Minecraft的红石，但在2d。

到目前为止，我有电源和电线。电源为电线供电，以后将用于为其他组件供电。

我面临的问题是当电线与电源断开时，它会保持通电状态，因为它的所有邻居都已通电。

以下是图片中所代表的问题：

导线未连接电源

电线已连接且已通电

断电，但导线仍处于活动状态

以下是来自电线的更新循环，显示：

def update(self):
    self.powered=False
    for b in blocks:
        #Find power sources/powered wires
        if not b.powered:
            continue
        adjacent = False
        if b.rect.collidepoint(self.rect.left,self.rect.top-5):
            adjacent = True
        if b.rect.collidepoint(self.rect.left,self.rect.bottom+5):
            adjacent = True
        if b.rect.collidepoint(self.rect.left-5,self.rect.top):
            adjacent = True
        if b.rect.collidepoint(self.rect.right+5,self.rect.top):
            adjacent = True
        if adjacent:
            self.powered = True
            break
    if not self.powered:
        pygame.draw.rect(screen, (0,0,0), self.rect, 0)
    else:
        pygame.draw.rect(screen, (200,0,0), self.rect, 0)

我需要电线由其他电线供电，但只有当它们都连接到电源时......所以我想，如果我可以递归检查每个电线段检查它连接到的所有电线，那么确定它连接到电源，然后我可以做到这一点。但我不知道该怎么做。

我尝试在线对象内部使用其他变量，例如self.sourced，表示线是否连接到电源，但这导致只有一根线能够接通电源。 （即使当电源断开时，电线也会停用，所以它有点工作......）

无论如何，如果有人知道如何做到这一点，请借给我你的知识。

以下是完整的源代码供参考：

import pygame
from pygame.locals import *

pygame.init()

screen=pygame.display.set_mode((640,480))
blocks=[]

class PowerSource(object):
    def __init__(self,pos):
        self.posx=pos[0]
        self.posy=pos[1]
        self.rect=pygame.Rect(self.posx,self.posy,32,32)
        self.powered=True
    def update(self):
        pygame.draw.rect(screen, (255,0,0), self.rect, 0)
    def repos(self):
        pass
class Circuit(object):
    def __init__(self,pos):
        self.powered=False
        self.posx=pos[0]
        self.posy=pos[1]
        self.rect=pygame.Rect(self.posx,self.posy,32,32)
    def update(self):
        self.powered=False
        for b in blocks:
            if not b.powered:
                continue
            adjacent = False
            if b.rect.collidepoint(self.rect.left,self.rect.top-5):
                adjacent = True
            if b.rect.collidepoint(self.rect.left,self.rect.bottom+5):
                adjacent = True
            if b.rect.collidepoint(self.rect.left-5,self.rect.top):
                adjacent = True
            if b.rect.collidepoint(self.rect.right+5,self.rect.top):
                adjacent = True
            if adjacent:
                self.powered = True
                break
        if not self.powered:
            pygame.draw.rect(screen, (0,0,0), self.rect, 0)
        else:
            pygame.draw.rect(screen, (200,0,0), self.rect, 0)
while True:
    place=1
    screen.fill((255,255,255))
    mse=pygame.mouse.get_pos()
    mse=((mse[0]/32)*32,(mse[1]/32)*32)
    pressed=pygame.mouse.get_pressed()
    if pressed==(1,0,0):
        pressed='L'
    elif pressed==(0,0,1):
        pressed='R'
    for b in blocks:
        b.update()
    pygame.draw.rect(screen, (255,0,0), (mse[0],mse[1],32,32), 2)
    for e in pygame.event.get():
        if e.type==QUIT:
            exit()
    key=pygame.key.get_pressed()
    if key[K_SPACE]:
        for b in blocks:
            if b.rect.collidepoint(mse):
                place=0
        if place==1:
            blocks.append(PowerSource(mse))
    if pressed=='L':
        for b in blocks:
            if b.rect.collidepoint(mse):
                place=0
        if place==1:
            blocks.append(Circuit(mse))

    elif pressed=='R':
        for b in blocks:
            if b.rect.collidepoint(mse):
                blocks.remove(b)
    pygame.display.flip()

Answer 1

我认为你因为一些事情而遇到麻烦。

您可能希望为您的块提供某种包含数据结构（另一个类，而不仅仅是列表），或者让您的块保持邻接列表。我认为后者可能更容易理解这个问题，虽然前者可能更适合你，因为你需要能够删除块（邻接列表有很多簿记）。这取决于你，但我选择邻接列表，因为它对我来说更容易。 ：）

一旦你失败了，你应该编写另一个方法作为你的递归入口点。它可以很简单。你可能需要做一些跟上你去过的地方的事情;假设有一个名为neighbors的邻接列表，我在这里做了一个作为方法参数的集合。

def is_connected_to_powered(self, visited=()):
    if self in visited:
        return False
    visited = set(visited)
    visited.add(self)
    for neighbor in self.neighbors:
        if neighbor.powered or neighbor.is_connected_to_powered(visited):
            return True
    return False

这回答了这个问题，但我也可以考虑从一个公共基类（Block？）派生Circuit和PowerSource，因为我怀疑有适当的OO可以帮助你顺利完成。

Answer 2

根据我的经验，简短的回答是向对象添加一个标志，指示它们是否已在递归算法中进行过测试。这是因为你可以有循环。因此，当您移除电源时，让它调用一个函数来“检查它所触及的所有块的源”。在查看是否有源作为邻居之前，每个块都会设置它自己的标记，表明“我已被检查过”。如果它有一个源作为邻居它返回“是我有一个源”。如果没有，那么它会检查是否有任何邻居有一个来源，但前提是它们尚未被标记为已被检查。然后它将查看是否有任何返回值是“是我有源”，如果是，则返回“是”如果不是，则返回“否”并关闭。