美团笔试回忆

最新推荐文章于 2024-09-29 16:46:50 发布

cjmHK

最新推荐文章于 2024-09-29 16:46:50 发布

阅读量466

点赞数

分类专栏：快速幂文章标签：算法

本文链接：https://blog.csdn.net/qq_40634175/article/details/108475589

版权

快速幂专栏收录该内容

2 篇文章 0 订阅

订阅专栏

前言

美团笔试是5道编程题，只做出来3.5道（当时赶时间上学校，只做了一个小时二十分钟就交了，也可能是题目有点难，毕竟字节笔试只做了半个小时就AC了。。。），前两题是水题，就不说了。这里说一下比较有意思的第三题------可能是运气好，在纸上画了十多分钟就出思路了。

题意

令 $b_{i}=a_{i} \oplus 1\%i \oplus 2\%i \oplus ... \oplus n\%i$ ，要求 $b_{1} \oplus b_{2} \oplus ... \oplus b_{n}$ ，其中 $n<10^{5}$ ， $a_{i}$ 题中会给出。

思路

要求的结果无非是如下数字的异或。
$a_{1}\oplus a_{2}\oplus ...\oplus a_{n}$
$1\%1\oplus 2\%1\oplus ...\oplus n\%1$
$1\%2\oplus 2\%2\oplus ...\oplus n\%2$
$. . .$
$1\%n\oplus 2\%n\oplus ...\oplus n\%n$

化简后如下。
$a_{1}\oplus a_{2}\oplus ...\oplus a_{n}$
$0\oplus 0\oplus ...\oplus 0$
$1\oplus 2\oplus 0\oplus 1\oplus 2\oplus 0\oplus ...\oplus n\%2$
$1\oplus 2\oplus 3\oplus 0\oplus 1\oplus 2\oplus 3\oplus 0\oplus ...\oplus n\%3$
$. . .$
$1\oplus 2\oplus ...\oplus 0$

第一个式子直接累计异或即可，主要如何求下面的式子怎么求（对下面的式子从1开始编号）。
可以看出下面的式子是有循环节的，我们不如将其分组。
为了方便说明，假定 $pre_{i}=1\oplus 2\oplus ... \oplus i$ 。
比如对于1个式子，每组一个数0，最终结果为 $0^{n}$ （注意这个幂代表n个0异或）。
对于第2个式子，每组2个数1和0，最终结果为 $1^{\frac{n}{2}}\oplus pre_{n\%2}$ 。
一般地，对于第i个式子，每组n个数1,2,…,i-1,0，最终结果为 $pre_{i-1}^{\frac{n}{i}}\oplus pre_{n\%i}$ 。
$pre_{i-1}^{\frac{n}{i}}$ 可以用快速幂求（注意不是常规的快速幂，要将乘法转化为异或）。

时间复杂度分析

一共有n个式子，求每个式子需要快速幂，时间复杂度为 $\mathcal{O}(\log(n))$ ，总的时间复杂度为 $\mathcal{O}(n\log(n))$ 。

实现

#include<iostream>
using namespace std;
const int N=1e5+10;
int a[N],pre[N];
int qpow(int a,int b){
    int res=0,rem=a;
    while(b){
        if(b&1) res^=rem;
        rem^=rem;
        b>>=1;
    }
    return res;
}
int main(){
    for(int i=1;i<N;i++){
        pre[i]=i^pre[i-1];
    }
    int n;cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    int res=0;
    for(int i=1;i<=n;i++){
        res^=a[i];
    }
    for(int i=1;i<=n;i++){
        res^=qpow(pre[i-1],n/i)^pre[n%i];
    }
    cout<<res<<endl;
    return 0;
}