hello-algo/codes/python/modules/vertex.py

# File: vertex.py
# Created Time: 2023-02-23
# Author: Krahets (krahets@163.com)

from typing import List

class Vertex:
    """ 顶点类 """
    def __init__(self, val: int) -> None:
        self.val = val

def vals_to_vets(vals: List[int]) -> List['Vertex']:
    """ 输入值列表 vals ，返回顶点列表 vets """
    return [Vertex(val) for val in vals]

def vets_to_vals(vets: List['Vertex']) -> List[int]:
    """ 输入顶点列表 vets ，返回值列表 vals """
    return [vet.val for vet in vets]