洛谷 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


基本思路:

树状数组维护两个前缀。
0前缀:1~k前k个位置的总种类数。
1前缀:1~k前k个位置的右端点个数。
那么每次询问[L,R],ans = c[0][R] - c[1][L-1];

为什么是这样呢?
因为每一个R节点(右端点)都代表着一个雷的结束。
我们可以求出前L-1个位置已经结束的雷的个数(c[1][L-1])
而到R节点一共有c[0][R]种雷,在L-1前面这些位置结束了c[1][L-1]个。
所以在区间[L,R]区间内雷的种类个数为c[0][R] - c[1][L-1]个。


具体实现代码:

int n,m,t[2][maxn],order,L,R;

void addpoint(int primpos,int orde)
{
  for(int i = primpos;i<=n;i = i+(i&-i))
  {
    t[orde][i]++;
  }
  return;
}
int query(int primpos,int orde)
{
  int res = 0;
  for(int i= primpos;i;i = i-(i&-i))
    res = res + t[orde][i];
  return res;
}

int main()
{
  freopen("War.in","r",stdin);
  n = gi(); m = gi();
  while(m--)
    {
        order = gi();L = gi(); R = gi();
        if(order == 1)
      {
        addpoint(L,0);
        addpoint(R,1);
      }
    else if(order == 2)
      printf("%d\n",query(R,0) - query(L-1,1));
    }return 0;
}

。。。。。。。

你可能感兴趣的:(树状数组)