CH3801Rainbow的信号

Description

Freda发明了传呼机之后，rainbow进一步改进了传呼机发送信息所使用的信号。由于现在是数字、信息时代，rainbow发明的信号用N个自然数表示。为了避免两个人的对话被大坏蛋VariantF偷听T_T，rainbow把对话分成A、B、C三部分，分别用a、b、c三个密加密码。现在Freda接到了rainbow的信息，她的首要工作就是解密。Freda了解到，这三部分的密码计算方式如下：
在1~N这N个数中，等概率地选取两个数l、r，如果l>r，则交换l、r。把信号中的第l个数到第r个数取出来，构成一个数列P。
A部分对话的密码是数列P的xor和的数学期望值。xor和就是数列P中各个数异或之后得到的数； xor和的期望就是对于所有可能选取的l、r，所得到的数列的xor和的平均数。
B部分对话的密码是数列P的and和的期望，定义类似于xor和。
C部分对话的密码是数列P的or和的期望，定义类似于xor和。

代码改自紫书
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int a[1000001], b[1000001], n;
double ansxor, ansand, ansor;

inline void solve(int k){
    register int last[2]={0,0},c1=0,c2=0;
    for(register int i=1;i<=n;i++) {//处理b[i]和length=1的情况
        b[i]=((a[i]>>k)&1);//取出每个数的第k位
        if(b[i]){//当且仅当b[i]!=0时才会有贡献
            ansxor+=(1<<k)*1.0/n/n;
            ansand+=(1<<k)*1.0/n/n;
            ansor+=(1<<k)*1.0/n/n;
        }
    }
    for(register int i=1;i<=n;i++){//处理length>=2的情况
        if(b[i]==0){
            ansor+=(1<<k)*2.0/n/n*last[1];
            ansxor+=(1<<k)*2.0/n/n*c2;
        }else{
            ansand+=(1<<k)*2.0/n/n*(i-1-last[0]);//(i-1)-(last[0]+1)+1
            ansor+=(1<<k)*2.0/n/n*(i-1);
            ansxor+=(1<<k)*2.0/n/n*c1;
        }
        c1++;
        if(b[i])swap(c1,c2);
        last[b[i]]=i;
    }
}
int main(){
    cin>>n;
    for(register int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(register int i=0;i<31;i++)solve(i);
    printf("%.3f %.3f %.3f\n",ansxor,ansand,ansor);
}

转载于:https://www.cnblogs.com/lqhsr/p/10808065.html