字符匹配 835. Trie字符串统计

835. Trie字符串统计

维护一个字符串集合，支持两种操作：

I x 向集合中插入一个字符串 x；
Q x 询问一个字符串在集合中出现了多少次。
共有 N 个操作，输入的字符串总长度不超过 105，字符串仅包含小写英文字母。

输入格式
第一行包含整数 N，表示操作数。

接下来 N 行，每行包含一个操作指令，指令为 I x 或 Q x 中的一种。

输出格式
对于每个询问指令 Q x，都要输出一个整数作为结果，表示 x 在集合中出现的次数。

每个结果占一行。

数据范围

1≤N≤2∗104

输入样例：

5
I abc
Q abc
Q ab
I ab
Q ab

输出样例：

1
0
1

思路：字典树匹配，比较简单

class Node:
    def __init__(self,c):
        self.c = c
        self.times = 0
        self.next = [None]*26

N = int(input())
root = Node("")
for a in range(N):
    q,s = tuple(input().split())
    if(q == 'I'):
        cur = root
        for c in s:
            if cur.next[ord(c)-97]!=None:
                cur = cur.next[ord(c)-97]
            else:
                aaa = Node(c) 
                cur.next[ord(c)-97] = aaa
                cur = cur.next[ord(c)-97]
        cur.times = cur.times+1
    if(q == 'Q'):
        cur = root
        for c in s:
            cur = cur.next[ord(c)-97]  
            if(cur==None):
                break
        if(cur!=None):
            print(cur.times)
        else:
            print(0)

143. 最大异或对

在给定的 N 个整数 A1，A2……AN 中选出两个进行 xor（异或）运算，得到的结果最大是多少？

输入格式
第一行输入一个整数 N。

第二行输入 N 个整数 A1～AN。

输出格式
输出一个整数表示答案。

数据范围

1≤N≤105,
0≤Ai<231

输入样例：

3
1 2 3

输出样例：

3

思路：先思考朴素做法，假设,每一个字符串都要与其他字符串进行亦或运算然后再比较大小，O(N^2). 用字典树进行优化，相当于一个字符串只需要与字典树进行匹配一次就可以了，所以有N个字符串，匹配N次，时间复杂度为 O(N).
字典树匹配思路，尽量找每个节点为异或值为1的节点。从高位到低位进行比较，思路还是比较简单的。
如果数据长度不一样，要变成同长，前面补0.

代码：

class Node:
    def __init__(self):
        self.next = [None]*2

_ = input()
data = input().split()
# 使其全部等长
maxLen = 0
for index,item in enumerate(data):
    data[index] = bin(int(item))[2:]
    if len(data[index]) > maxLen:
        maxLen = len(data[index])
data = [item.rjust(maxLen,'0') for item in data]

root = Node()

for s in data:
    cur = root
    for c in s:
        ic = int(c)
        if(cur.next[ic]==None):
            cur.next[ic] = Node()
            cur = cur.next[ic]
        else:
            cur = cur.next[ic]

max = "0".rjust(maxLen,'0')


for s in data:
    cur = root
    curs = "" # 存储当前得出的异或值
    for c in s:
        if cur.next[0]!=None and cur.next[1]!=None:    
            if int(c) ^ 1 == 1: 
                cur = cur.next[1]
                curs = curs+'1'
            else:
                cur = cur.next[0]
                curs = curs+'1'
        else:
            if cur.next[0] !=None:
                cur = cur.next[0]
                curs = curs+str(int(c)^0)
            else:
                cur = cur.next[1]
                curs = curs+str(int(c)^1)
    if curs > max:
        max = curs
print(int(max,2))