Editing 1185 - Cub2 (section)

= Explicație =
'''Atenție! Pentru acest test se rezolvă doar cerința 2).'''

<code>4</code> – becuri inscripționate cu numere prime pe fața 1: <code>2, 3, 17, 19</code>

<code>3</code> – becuri inscripționate cu numere prime pe fața 2: <code>3, 5, 23</code>

<code>4</code> – becuri inscripționate cu numere prime pe fața 3: <code>5, 7, 13, 23</code>

<code>3</code> – becuri inscripționate cu numere prime pe fața 4: <code>7, 11, 19</code>

<syntaxhighlight lang="python" line="1">
def generate_spiral_layer(n, start_val):
    """ Generate a spiral layer of size n x n starting from start_val """
    layer = [[0] * n for _ in range(n)]
    num = start_val
    for layer_idx in range((n + 1) // 2):
        for i in range(layer_idx, n - layer_idx):
            layer[layer_idx][i] = num
            num += 1
        for i in range(layer_idx + 1, n - layer_idx):
            layer[i][n - layer_idx - 1] = num
            num += 1
        for i in range(n - layer_idx - 2, layer_idx - 1, -1):
            layer[n - layer_idx - 1][i] = num
            num += 1
        for i in range(n - layer_idx - 2, layer_idx, -1):
            layer[i][layer_idx] = num
            num += 1
    return layer

def generate_cube(N):
    """ Generate a cube of size N x N x N filled in spiral order """
    cube = [[[0] * N for _ in range(N)] for _ in range(N)]
    num = 1
    for z in range(N):
        layer = generate_spiral_layer(N, num)
        for y in range(N):
            for x in range(N):
                cube[z][y][x] = layer[y][x]
        num += N * N
    return cube

def find_coordinates(cube, V):
    """ Find coordinates (x, y, z) of the value V in the cube """
    N = len(cube)
    for z in range(N):
        for y in range(N):
            for x in range(N):
                if cube[z][y][x] == V:
                    return (x + 1, y + 1, z + 1)
    return None

def count_lights_on_faces(cube):
    """ Count the number of lights on each face of the cube """
    N = len(cube)
    counts = {'front': 0, 'back': 0, 'left': 0, 'right': 0, 'top': 0, 'bottom': 0}
    
    # Counting lights on each face
    for z in range(N):
        counts['top'] += N
        counts['bottom'] += N
    
    for y in range(N):
        counts['front'] += N
        counts['back'] += N
    
    for x in range(N):
        counts['left'] += N
        counts['right'] += N
    
    return counts

# Example Usage
N = 3
V = 7  # Example value to find coordinates for
cube = generate_cube(N)

coordinates = find_coordinates(cube, V)
face_counts = count_lights_on_faces(cube)

print(f"Coordinates of the light with number {V}: {coordinates}")
print(f"Number of lights on each face: {face_counts}")
</syntaxhighlight>