[C++&Rust]LeetCode No.1486 数组异或操作(每日一题)

原贴地址：http://blog.leanote.com/post/dawnmagnet/c8deab0f58d7

题目

思路分析

这个题目就算我们暴力做，也只是O(n)的复杂度，而且题目的数据给的也不是很大，暴力一定非常快。（窃喜
于是我们暴力做了一下，很快啊，就出来了，直接就AC了，还只用了0ms。

impl Solution {
    pub fn xor_operation(n: i32, mut start: i32) -> i32 {
        let mut res = 0;
        for i in 0..n {
            res ^= (start + 2 * i);
        }
        res
    }
}

那么请问我们还有继续往下优化的必要吗？
当然有。
往下优化，复杂度可能是O(1)或者O(lgn)，一般来说，我们要有基本的概念。
怎么样才能让复杂度到达O(1)呢？一般来说是答案是有规律的，套个公式什么的直接就能出答案。
怎么样才能让复杂度到达O(lgn)呢？一般来说是通过除2来构造的，除2之前之后获取一些信息，再组合组合。这样才能得到O(lgn)的答案。

给大家一些做题目时候的小tips，遇到这种数列找规律的题目，先打印出来看看，是否符合规律（基本上都能肉眼观察出来）

以下是初始值为0的时候前20项
0 2 6 0 8 2 14 0 16 2 22 0 24 2 30 0 32 2 38 
以下是初始值为1的时候前20项
1 2 7 0 9 2 15 0 17 2 23 0 25 2 31 0 33 2 39 
以下是初始值为2的时候前20项
2 6 0 8 2 14 0 16 2 22 0 24 2 30 0 32 2 38 0 
以下是初始值为3的时候前20项
3 6 1 8 3 14 1 16 3 22 1 24 3 30 1 32 3 38 1 

不难看出，以任何数字为初始值开始都是有规律可循的。而且每一个都是4项为一组。

初始值	规律
0	* 2 * 0
1	* 2 * 0
2	2 * 0 *
3	3 * 1 *

不难看出都是有规律可循的，而只要定位了这些规律点，就非常好求每一项了，比如求起始为0的第五项，我们只需要在里面找到第五项的位置，就是2左边的位置，再用2去异或2 * n 也就是2 ^ 10 = 8，按照这个方法，只要我们确定了特征点，在去反推任何位置的数字，只需要O(1)的时间，最多不会超过三步

• 对4取余，在特征数组中定位
• 如果是定值，则返回定值
• 如果不是定值，则拿定值反推

OK，就好了。
而对于数组的预处理，我们只需要定一个长度为4的特征数组，从第一项循环到第八项，就可以确定这个数组里哪些数字是不变的常量，我们就用常量表示，哪些数字是不确定的，我们就将其标注为-1（与表格中的*等价）。
整个算法的复杂度是O(1)级别的，非常快速也非常简单易懂。

Rust代码

impl Solution {
    pub fn xor_operation(mut n: i32, mut start: i32) -> i32 {
        let mut sto = [-1; 4];
        let mut tmp = 0;
        for i in 0..4 {
            tmp ^= (start + 2 * i);
            sto[i as usize] = tmp;
        }
        for i in 0..4 {
            tmp ^= (start + 2 * (i + 4));
            if sto[i as usize] != tmp {
                sto[i as usize] = -1;
            }
        }
        let n = n as usize;
        match sto[(n - 1) % 4] {
            -1 => sto[n % 4] ^ (start + 2 * n as i32),
            _ => sto[(n - 1) % 4]
        }
    }
}

C++代码

class Solution {
public:
    int xorOperation(int n, int start) {
        vector<int> sto = {0, 0, 0, 0};
        int tmp = 0;
        for (int i = 0; i < 4; ++i) {
            tmp ^= (start + 2 * i);
            sto[i] = tmp;
        }
        for (int i = 0; i < 4; ++i) {
            tmp ^= (start + 2 * (i + 4));
            if (sto[i] != tmp)
                sto[i] = -1;
        }
        if (sto[(n - 1) % 4] == -1) 
            return sto[n % 4] ^ (start + 2 * n);
        else 
            return sto[(n - 1) % 4];
    }
};