# -*- coding: utf-8 -*-
from typing import Any


class Tree(object):
    """
    樹-類數(shù)組模塊
    """
    __tree = []

    def __init__(self) -> None:
        """
        樹對象
        """
        pass

    def add_arrays(self, *arrays) -> None:
        """
        添加數(shù)組的方法
        :param arrays: 數(shù)組
        :return:
        """
        if arrays and len(arrays) > 0:
            for arr in arrays:
                self.__tree.append(arr)

    def add(self, obj) -> int:
        """
        添加對象的方法
        :param obj: 對象
        :return:
        """
        self.__tree.append(obj)
        return len(self.__tree)

    def to_array(self) -> list:
        """
        返回數(shù)組對象
        :return:
        """
        return self.__tree

    def filter(self) -> list:
        """
        當(dāng)前數(shù)組去重復(fù)方法
        :return:
        """
        self.__tree = list(set(self.__tree))

    def get(self, index: int) -> Any:
        """
        獲取對象的方法
        :param index:
        :return:
        """
        return self.__tree[index]

    def size(self) -> int:
        """
        獲取當(dāng)前數(shù)據(jù)的長度
        :return:
        """
        return len(self.__tree)

    def remove(self, index: int) -> None:
        """
        刪除數(shù)據(jù)的方法
        :param index:
        :return:
        """
        self.__tree.remove(self.__tree[index])

    def foreach(self, function) -> list:
        """
        for循環(huán)操作方法
        :param function: 傳遞到的方法
        :return:
        """
        result = []
        for arr in self.__tree:
            res = function(arr)
            if res:
                result.append(res)
        return result