【树状数组练习】洛谷 树状数组模板1、2 P1908 逆序对

洛谷 P3374 树状数组模板1

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <string>
//typedef long long int LL;
typedef std::pair<std::string ,char> PP;
const int N = 5e5+5;
int tree[N];
int n,m,vv,op,a,b;

void add(int i,int v)
{
   while(i<=n){
      tree[i] += v;
      i += i&(-i);
   }
}

int query(int i)
{
   int re = 0;
   while(i>=1){
      re += tree[i];
      i -= i&(-i);
   }
   return re;
}

int main()
{
   scanf("%d%d",&n,&m);
   for(int i=1; i<=n; i++){
      scanf("%d",&vv);
      add(i,vv);
   }
   for(int i=0; i<m; i++){
      scanf("%d%d%d",&op,&a,&b);
      if(op == 1){
         add(a,b);
      }
      else{
         printf("%d\n",query(b)-query(a-1));
      }
   }
   return 0;
}

洛谷 P3368树状数组模板2

在区间中求某一单点的树状数组。初始化树状数组时，增加的值为"此点比上一个点大多少"

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <string>
typedef long long int LL;
//typedef std::pair<std::string ,char> PP;
const int N = 5e5+5;
LL tree[N];
int n,m,op,a,b;
LL c,vv,ls;

void add(int i,LL v)
{
   while(i<=n){
      tree[i] += v;
      i += i&(-i);
   }
}

LL query(int i)
{
   LL re = 0;
   while(i>=1){
      re += tree[i];
      i -= i&(-i);
   }
   return re;
}

int main()
{
   scanf("%d%d",&n,&m);
   ls = 0;
   for(int i=1; i<=n; i++){
      scanf("%lld",&vv);
      add(i,vv-ls);
      ls = vv;
   }
   for(int i=0; i<m; i++){
      scanf("%d",&op);
      if(op == 1){
         scanf("%d%d%lld",&a,&b,&c);
         add(a,c);
         add(b+1,-c);
      }
      else{
         scanf("%d",&a);
         printf("%lld\n",query(a));
      }

   }
   return 0;
}

洛谷 P1908 逆序对

数据大小的范围是1e9，但是我们只需要知道他们的大小关系，所以可以压成5e5做

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
typedef long long int LL;
//typedef std::pair<int ,int> PP;
const int N = 5e5+5;
int tree[N],ord[N];
struct AA
{
   int val,n;
}qu[N];
int n,v,idx;
LL ans;

void add(int i,int x)
{
   while(i<=n){
      tree[i] += x;
      i += i&(-i);
   }
}

LL qry(int i)
{
   LL re = 0;
   while(i>0){
      re += tree[i];
      i -= i&(-i);
   }
   return re;
}

bool cmp(AA x , AA y)
{
   if(x.val == y.val) return x.n<y.n;
   return x.val < y.val;
}

int main()
{
   //freopen("D:\\EdgeDownloadPlace\\P1908_11.in","r",stdin);
   ans = 0;
   scanf("%d",&n);
   for(int i=1; i<=n; i++){
      scanf("%d",&qu[i].val);
      qu[i].n = i;
   }
   std::sort(qu+1,qu+n+1,cmp);
   for(int i=1; i<=n; i++){
      ord[qu[i].n] = i;
   }
   for(int i=1; i<=n; i++){
      add(ord[i],1);
      ans += i-qry(ord[i]);
   }
   printf("%lld\n",ans);

   return 0;
}