题目大意:
题目链接:https://jzoj.net/senior/#main/show/3518
题目图片:
http://wx2.sinaimg.cn/mw690/0060lm7Tly1fy2vsa8tmfj30jd0boaab.jpg
http://wx2.sinaimg.cn/mw690/0060lm7Tly1fy2vsa8oz7j30j309t3yi.jpg
给出一个数列,求 a i o r a i + 1 o r . . . o r a j ≤ m a_i\ or\ a_{i+1}\ or...or \ a_j\leq\ m ai or ai+1 or...or aj≤ m的 i , j i,j i,j个数。
思路:
很明显的,如果
a
i
o
r
.
.
.
o
r
a
j
≤
m
a_i\ or...or\ a_j\leq m
ai or...or aj≤m,那么
a
i
+
1
o
r
.
.
.
o
r
a
j
≤
m
a_{i+1}\ or...or\ a_j\leq\ m
ai+1 or...or aj≤ m。
那么就可以维护两个指针
i
,
j
i,j
i,j,表示
a
i
o
r
.
.
.
o
r
a
j
≤
m
a_i\ or...or\ a_j\leq m
ai or...or aj≤m,每次往后移一位
j
j
j,用
n
u
m
[
k
]
num[k]
num[k]表示从
a
i
a_i
ai到
a
j
a_j
aj中二进制下第
k
k
k位是
1
1
1的个数。将
n
u
m
num
num所表示的数和
m
m
m比较,如果还符合要求,那么
a
n
s
ans
ans就要加
j
−
i
j-i
j−i。因为
a
i
o
r
.
.
.
o
r
a
j
≤
m
a_i\ or...or\ a_j\leq m
ai or...or aj≤m,那么
a
i
+
1
o
r
.
.
.
o
r
a
j
≤
m
a_{i+1}\ or...or\ a_j\leq\ m
ai+1 or...or aj≤ m,
a
i
+
2
o
r
.
.
.
o
r
a
j
≤
m
a_{i+2}\ or...or\ a_j\leq\ m
ai+2 or...or aj≤ m,以此类推。
如果不符合要求,那么就将
i
i
i指针往后移。知道符合要求为止。
时间复杂度
O
(
n
l
o
g
n
)
O(n\ log_\ n)
O(n log n)
代码:
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
const int LG=31;
const int N=100100;
int n,m,i,j,a[N],num[LG+1];
ll ans;
bool check() //判断num大还是m大
{
int mm=0;
for (int i=LG;i>=1;i--)
mm+=((num[i]>0)*(1<<(i-1)));
return mm<m;
}
void write(ll x) //输出流,防止炸long long
{
if (x>9) write(x/10);
putchar(x%10+48);
}
int main()
{
freopen("evolve.in","r",stdin);
freopen("evolve.out","w",stdout);
scanf("%d%d",&n,&m);
for (i=1;i<=n;i++)
scanf("%d",&a[i]);
j=1;
i=0;
while (i<n)
{
i++;
for (int k=1;k<=LG;k++)
num[k]+=((a[i]&(1<<(k-1)))==(1<<(k-1)));
while (!check())
{
for (int k=1;k<=LG;k++)
num[k]-=((a[j]&(1<<(k-1)))==(1<<(k-1)));
j++;
}
ans+=(ll)max((i-j),0);
}
write(ans);
return 0;
}