bzoj 5092: 分割序列 贪心

题意

对于一个长度为n的非负整数序列b_1,b_2,…,b_n，定义这个序列的能量为：f(b)=max{i=0,1,…,n}((b_1 xor b_2 xor…xor b_i)+(b_{i+1} xor b_{i+2} xor…xor b_n))其中xor表示按位异或(XOR)，给定一个长度为n的非负整数序列a_1,a_2,…,a_n，请计算a的每个前缀的能量值。
n<=300000,0<=a_i<=10^6

分析

首先我们设a[i]表示b的异或前缀和，那么我们就是要对于每个i，求max(a[j]+a[i] xor a[j])。
从高位到低位贪心地想，如果a[i]的第k位是1，那么无论a[j]的这一位是什么，对答案的贡献都是2^k。
若a[i]的第k位是0，那么如果a[j]这一位是1，贡献就是2*2^k，否则贡献为0。
那么我们可以先预处理出tim[s]表示最小的i使得a[i]&s=s，然后对a[i]从高位到低位进行贪心，看每一位0是否能找到一个j使得a[j]的这一位是1即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define MIN(x,y) x=min(x,y)
using namespace std;

const int N=300005;
const int inf=1000000000;

int n,bin[25],tim[1048580],a[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int main()
{
    bin[0]=1;
    for (int i=1;i<=20;i++) bin[i]=bin[i-1]*2;
    for (int i=1;i<bin[20];i++) tim[i]=inf;
    n=read();
    for (int i=1;i<=n;i++) a[i]=read()^a[i-1],MIN(tim[a[i]],i);
    for (int i=bin[20]-1;i>=1;i--)
    {
        if (tim[i]==inf) continue;
        for (int j=0;j<20;j++)
            if (i&bin[j]) MIN(tim[i-bin[j]],tim[i]);
    }
    for (int i=1;i<=n;i++)
    {
        int ans=0,now=0;
        for (int j=19;j>=0;j--)
            if (a[i]&bin[j]) ans+=bin[j];
            else if (tim[now+bin[j]]<=i) now+=bin[j],ans+=bin[j]*2;
        printf("%d\n",ans);
    }
    return 0;
}