배열 에서 두 개의 차이 나 m 이상 의 개수

정수 m 와 n 의 각 숫자 A1, A2,....................................................................................
1. aDigit = 1, mDigit = 1 일 때 사전 의 k 위 는 0 이 고, 이 또는 결 과 는 1 이 므 로 계속 검색 해 야 합 니 다. k 위 는 1 이 고, 이 또는 결 과 는 0 이 며, mDigit 보다 작 으 므 로 신경 쓰 지 마 십시오.
2. aDigit = 0, mDigit = 1 일 때 사전 의 k 위 는 1 이 고, 이 또는 결 과 는 1 이 므 로 계속 검색 해 야 합 니 다. k 위 는 0 이 고, 이 또는 결 과 는 0 이 며, mDigit 보다 작 으 므 로 신경 쓰 지 마 십시오.
3. aDigit = 1, mDigit = 0 일 때 사전 의 k 위 는 0 이 고, 이 또는 결 과 는 1 이 며, 대응 하 는 분기 의 모든 수 와 다 르 거나, 결 과 는 m 보다 크 고, k 위 는 1 이 며, 이 또는 결 과 는 0 이 며, 재 귀적 으로 결 과 를 얻는다.
4. aDigit = 0, mDigit = 0 시,
사전 에서 k 위 는 1 이 고, 이 또는 결 과 는 1 이 며, 대응 하 는 분기 의 모든 수 와 다 르 거나, 결 과 는 m 보다 크 고, k 위 는 0 이 며, 이 또는 결 과 는 0 이 며, 재 귀 는 결 과 를 얻 을 수 있 습 니 다.

public class Main 
{
    private static class TrieTree 
    {
        TrieTree[] next = new TrieTree[2];
        int count = 1;
    }
    private static long solve(int[] a, int m) 
    {
        TrieTree trieTree = buildTrieTree(a);
        long result = 0;
        for (int i = 0; i < a.length; i++) 
        {
            result += queryTrieTree(trieTree, a[i], m, 31);
        }
        return result / 2;
    }
    private static long queryTrieTree(TrieTree trieTree, int a, int m, int index) 
    {
        if(trieTree == null)
            return 0;
        TrieTree current = trieTree;
        for (int i = index; i >= 0; i--) 
        {
            int aDigit = (a >> i) & 1;
            int mDigit = (m >> i) & 1;
            if(aDigit == 1 && mDigit == 1) 
            {
                if(current.next[0] == null)
                    return 0;
                current = current.next[0];
            } 
            else if (aDigit == 0 && mDigit == 1) 
            {
                if(current.next[1] == null)
                    return 0;
                current = current.next[1];
            } 
            else if (aDigit == 1 && mDigit == 0) 
            {
                long p = queryTrieTree(current.next[1], a, m, i - 1);
                long q = current.next[0] == null ? 0 : current.next[0].count;
                return p + q;
            } 
            else if (aDigit == 0 && mDigit == 0) 
            {
                long p = queryTrieTree(current.next[0], a, m, i - 1);
                long q = current.next[1] == null ? 0 : current.next[1].count;
                return p + q;
            }
        }
        return 0;
    }
    private static TrieTree buildTrieTree(int[] a) 
    {
        TrieTree trieTree = new TrieTree();
        for (int i = 0; i < a.length; i++) 
        {
            TrieTree current = trieTree;
            for (int j = 31; j >= 0; j--) 
            {
                int digit = (a[i] >> j) & 1;
                if(current.next[digit] == null) 
                {
                    current.next[digit] = new TrieTree();
                } 
                else 
                {
                    current.next[digit].count ++;
                }
                current = current.next[digit];
            }
        }
        return trieTree;
    }
}