Drop some Blox! The one who drops in the most efficient way wins! 🏆
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
import dataclasses |
|
|
from dataclasses import dataclass |
|
|
from typing import Dict, List, Optional |
|
|
|
|
|
import numpy as np |
|
|
|
|
|
color_mapper = { |
|
|
"Y": "🟨", |
|
|
"G": "🟩", |
|
|
"B": "🟦", |
|
|
"R": "🟥", |
|
|
"P": "🟪", |
|
|
"O": "🟧", |
|
|
"W": "⬜", |
|
|
} |
|
|
|
|
|
|
|
|
@dataclass(frozen=True) |
|
|
class Block: |
|
|
block_id: int |
|
|
width: int |
|
|
height: int |
|
|
color: str |
|
|
values: np.array # Note: first row is the bottom, values are flipped when printing the result |
|
|
|
|
|
def __post_init__(self): |
|
|
if self.width != len(self.values[0]) or self.height != len(self.values): |
|
|
raise Exception( |
|
|
f"This block with width {self.width}, height: {self.height} and values {self.values} does not exist" |
|
|
) |
|
|
|
|
|
def __repr__(self): |
|
|
result_str = "Block:\n" |
|
|
for row in reversed(self.values): |
|
|
for item in row: |
|
|
result_str += color_mapper[item] if item else "⬜️" |
|
|
result_str += "\n" |
|
|
return result_str |
|
|
|
|
|
|
|
|
@dataclass(frozen=True) |
|
|
class Rewards: |
|
|
points: Dict[str, int] |
|
|
multiplication_factors: Dict[str, int] |
|
|
|
|
|
|
|
|
@dataclass() |
|
|
class Field: |
|
|
width: int |
|
|
height: int |
|
|
values: Optional[ |
|
|
np.ndarray |
|
|
] = None # Note: first row is the bottom, values are flipped when printing the result |
|
|
|
|
|
def __post_init__(self): |
|
|
self.values = np.empty((self.height, self.width), dtype=str) |
|
|
self.values[:] = "" |
|
|
|
|
|
def __repr__(self): |
|
|
result_str = "⬛️" * (self.width + 2) + "\n" |
|
|
for row in reversed(self.values): |
|
|
result_str += "⬛️" |
|
|
for item in row: |
|
|
result_str += color_mapper[item] if item else "⬜" |
|
|
result_str += "⬛️\n" |
|
|
result_str += "⬛️" * (self.width + 2) |
|
|
return result_str |
|
|
|
|
|
def clear_field(self): |
|
|
self.values[:] = "" |
|
|
|
|
|
|
|
|
@dataclass() |
|
|
class Solution: |
|
|
"""A solution comprises two lists that describe which blocks are dropped in what order and where |
|
|
- order: order of the list |
|
|
- which blocks: determined by block_ids |
|
|
- where: most left part of the block is dropped at block_position |
|
|
""" |
|
|
|
|
|
block_ids: List[int] = dataclasses.field(default_factory=list) |
|
|
block_positions: List[int] = dataclasses.field(default_factory=list)
|
|
|
|
