为DNA序列创建数组散列

class DNAHashArray:
    def __init__(self, size):
        self.size = size
        self.array = [None] * size

    def hash_function(self, sequence):
        hash_value = 0
        for nucleotide in sequence:
            if nucleotide == 'A':
                hash_value += 1
            elif nucleotide == 'T':
                hash_value += 2
            elif nucleotide == 'C':
                hash_value += 3
            elif nucleotide == 'G':
                hash_value += 4
        return hash_value % self.size

    def insert(self, sequence):
        index = self.hash_function(sequence)
        if self.array[index] is None:
            self.array[index] = [sequence]
        else:
            self.array[index].append(sequence)

    def search(self, sequence):
        index = self.hash_function(sequence)
        if self.array[index] is not None:
            return sequence in self.array[index]
        return False

# 示例使用
dna_hash_array = DNAHashArray(100)
dna_hash_array.insert("ATCG")
dna_hash_array.insert("TAGC")
print(dna_hash_array.search("ATCG"))  # 输出: True
print(dna_hash_array.search("GGCC"))  # 输出: False