题目地址:点击打开链接
帅气的HYC经常早晨去锻炼(多么好的习惯~。有一天,他看到一路上的露珠,心里便产生了一个问题:一路上假如有N棵草,每颗草上可能会有露珠,或者没有露珠。连续的露珠会和为一体(>=2),并变为珍珠。
比如第1棵草上有露珠,第2棵草也有露珠。那么就会形成一个珍珠。第1棵草上有露珠,第2棵草有露珠,第3棵也有露珠,那么也会形成一个珍珠。相反,如果第1颗草有露珠,第2棵草没有露珠,就不会形成珍珠。
现在给你HYC今天早晨一路上的看到的N棵草,以及上面的露珠情况,给你Q次询问,每次询问[L,R],你能快速的告诉HYC,这个区间能形成多少个珍珠吗?ACCEPTIT!
第1行一个整数T,表示有T组测试数据。(T<=15)
第2行一个整数N, 表示有N棵草。(1<=N<=1000)
第3行有N个整数0或者1,中间用空格隔开, 1代表第i课草有露珠,0代表第i棵草没有露珠。
第4行是一个整数Q, 代表查询的次数。(1<=Q<=100000)
下面第(4+Q)行每行两个整数L, R。 代表查询的区间[L, R];(1<=L<=R<=N)
对于每组查询输出一行。
复制
1 4 1 1 0 1 3 1 2 2 3 1 4
1 0 1
思路:一串连续的露珠能形成一颗珍珠,一开始我们把这颗珍珠用树状数组记录在这串的最后一个位置上,每一次查
询通过树状数组查询这段区间的珍珠数,考虑下该区间的两端的露珠是否能形成珍珠。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 1e3+5;
int a[maxn], tree[maxn], n;
int lowbit(int x) { return x&-x; }
void update(int pos, int val)
{
while(pos < maxn)
{
tree[pos] += val;
pos += lowbit(pos);
}
}
int query(int pos)
{
int sum = 0;
while(pos)
{
sum += tree[pos];
pos -= lowbit(pos);
}
return sum;
}
int main(void)
{
int t;
cin >> t;
while(t--)
{
scanf("%d", &n);
for(int i = 1; i <= n; i++)
scanf("%d", &a[i]);
memset(tree, 0, sizeof(tree));
for(int i = 3; i <= n; i++)
if(a[i-2] && a[i-1] && !a[i])
update(i-1, 1);
if(a[n] && a[n-1]) update(n, 1);
int q, l, r;
scanf("%d", &q);
while(q--)
{
scanf("%d%d", &l, &r);
if(l == r) { puts("0"); continue; }
int ans = query(r)-query(l-1);
//因为下面两条是要在r-1、l+1的条件下使用的,所以l==r时不能用需要加特判
if(a[r] && a[r-1] && a[r+1]) ans++;
if(a[l-1] && a[l] && !a[l+1]) ans--;
printf("%d\n", ans);
}
}
return 0;
}