洛谷P2184 贪婪大陆(线段树/树状数组+括号覆盖)
题目背景
面对蚂蚁们的疯狂进攻,小FF的Tower defence宣告失败……人类被蚂蚁们逼到了Greed Island上的一个海湾。现在,小FF的后方是一望无际的大海, 前方是变异了的超级蚂蚁。 小FF还有大好前程,他可不想命丧于此, 于是他派遣手下最后一批改造SCV布置地雷以阻挡蚂蚁们的进攻。
题目描述
小FF最后一道防线是一条长度为N的战壕, 小FF拥有无数多种地雷,而SCV每次可以在[ L , R ]区间埋放同一种不同于之前已经埋放的地雷。 由于情况已经十万火急,小FF在某些时候可能会询问你在[ L' , R'] 区间内有多少种不同的地雷, 他希望你能尽快的给予答复。
对于30%的数据: 0<=n, m<=1000;
对于100%的数据: 0<=n, m<=10^5.
输入输出格式
输入格式:
第一行为两个整数n和m; n表示防线长度, m表示SCV布雷次数及小FF询问的次数总和。
接下来有m行, 每行三个整数Q,L , R; 若Q=1 则表示SCV在[ L , R ]这段区间布上一种地雷, 若Q=2则表示小FF询问当前[ L , R ]区间总共有多少种地雷。
输出格式:
对于小FF的每次询问,输出一个答案(单独一行),表示当前区间地雷总数。
输入输出样例
输入样例#1:
5 4 1 1 3 2 2 5 1 2 4 2 3 5
输出样例#1:
1 2
题解:
用树状数组维护区间左括号和右括号的个数,每一次询问时【l,r】的答案就是【1,r】的右括号的个数-【1,l-1】的左括号的个数;
代码:
#include<cstdio> #include<algorithm> using namespace std; template<class T>inline void read(T &x) { x=0;int f=0;char ch=getchar(); while(ch<'0'||ch>'9')f|=(ch=='-'),ch=getchar(); while(ch<='9'&&ch>='0')x=(x<<1)+(x<<3)+(ch^48),ch=getchar(); x=f?-x:x; return; } int n,m,s1[100010],s2[100010]; int lowbit(int x){return x&(-x);} void add1(int x) { while(x<=n) { s1[x]++; x+=lowbit(x); } } void add2(int x) { while(x<=n) { s2[x]++; x+=lowbit(x); } } int sum1(int x) { int ans=0; while(x>0) { ans+=s1[x]; x-=lowbit(x); } return ans; } int sum2(int x) { int ans=0; while(x>0) { ans+=s2[x]; x-=lowbit(x); } return ans; } int main() { int Q,l,r; read(n);read(m); for(int i=1;i<=m;i++) { read(Q);read(l);read(r); if(Q==1){ add1(l);add2(r); } else printf("%d\n",sum1(r)-sum2(l-1)); } return 0; }