题意
给出一个长度为n的序列a[],问你有多少个区间[l,r]满足a[l] or a[l+1] or … or a[r]>max(a[l],a[l+1],…,a[r])。
n<=200000,a[i]<=10^9
分析
假设我们要处理一个区间[l,r]内有多少满足题目条件的区间,我们可以先找到这个区间的最大值,设为mid。只要我们找到mid左边的第一个数,使得这个数or上a[mid]>a[mid],设其位置为L,同理找到mid右边的第一个R。那么左右端点均在[L+1,R-1]内的区间均为不满足区间。
然后分治[l,mid-1]和[mid+1,r]两个区间即可。
问题在于如何快速找到L和R。
我们可以对于每一位二进制位开一个数组c,按下标顺序记录所有该二进制位为1的数,同时预处理处对于每个位置,在该数组内最后一个不大于该位置的位置。
那么在找到mid后,我们就可以枚举a[mid]的每个0二进制位,然后找到mid左边和右边第一个该位为1的数即可。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
const int N=200005;
int n,rmq[N][20],lg[N],a[N],bin[40],c[40][N],low[40][N],tot[40];
LL ans;
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;
}
void get_rmq()
{
for (int i=1;i<=n;i++) rmq[i][0]=i;
for (int j=1;j<=lg[n];j++)
for (int i=1;i+bin[j]-1<=n;i++)
rmq[i][j]=a[rmq[i][j-1]]>a[rmq[i+bin[j-1]][j-1]]?rmq[i][j-1]:rmq[i+bin[j-1]][j-1];
}
int get_mx(int l,int r)
{
int w=lg[r-l+1];
return a[rmq[l][w]]>a[rmq[r-bin[w]+1][w]]?rmq[l][w]:rmq[r-bin[w]+1][w];
}
void solve(int l,int r)
{
if (l>r) return;
int mid=get_mx(l,r),L=l-1,R=r+1;
for (int i=0;i<=30;i++)
if (!(a[mid]&bin[i]))
{
if (low[i][mid]) L=max(L,c[i][low[i][mid]]);
if (low[i][mid]<tot[i]) R=min(R,c[i][low[i][mid]+1]);
}
ans-=(LL)(mid-L)*(R-mid);
solve(l,mid-1);solve(mid+1,r);
}
int main()
{
bin[0]=1;
for (int i=1;i<=30;i++) bin[i]=bin[i-1]*2;
n=read();
for (int i=1;i<=n;i++) a[i]=read(),lg[i]=log(i)/log(2);
get_rmq();
for (int i=0;i<=30;i++)
for (int j=1;j<=n;j++)
if (a[j]&bin[i]) c[i][++tot[i]]=j,low[i][j]=tot[i];
else low[i][j]=tot[i];
ans=(LL)n*(n+1)/2;
solve(1,n);
printf("%I64d",ans);
return 0;
}