一、线段树:线段树可以用于指定区间求和,区间求最大值,区间修改,单点修改等。
1 先设立树状结构体
在这里插入代码片struct Node{
int r,l;
int nsum;
}segTree[MAXN*4]
2 线段树初始化
void Bulid(int i,int r,int l){
seqTree[i].l=l;
seqTree[i].r=r;
int mid = (seqTree[i].l+seqTree[i].r)>>1;
Build(i<<1,r,mid);
Build(i<<1|1,mid+1,l);
seqTree[i].nsum=seqTree[i<<1].nsum+seqTree[i<<1|1].nsum;//上一层等于其分支 之和
}
3 求某特定区间的和 (递归型求法)
int Query(int i,int l,int r){
if(seqTree[i].r==r&&seqTree[i].l==l)
return seqTree[i].nsum;
int mid = (seqTree[i].l+seqTree[i].r)>>1;
if(r<=mid)
Query(i<<1,l,mid);
else If(l>mid)
Query(i<<1|1,mid+1,r);
else Query(i<<1,l,mid)+Query(i<<1|1,mid+1,r);
}
4 单点修改(t为特定项b为修改值b为正则为加 反之为减)
void Add(int i,int t,int b){
seqTree[i].nsum+=b; //该叶节点的父亲节点的值都要修改
if(seqTree[i].r==t&&seqTree[i].l==t)//找到该叶节点的时候结束
return;
int mid = (seqTree[i].l+seqTree[i].r)>>1;
if(r<=mid)
Add(i<<1,t,b);
else
Add(i<<1|1,t,b);
}
hud1556题
/*
HDU 1166 敌兵布阵
*/
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<iostream>
using namespace std;
const int MAXN=50005;
struct Node
{
int l,r;
int nSum;
}segTree[MAXN*4];
int num[MAXN];
void Build(int i,int l,int r)
{
segTree[i].l=l;
segTree[i].r=r;
if(l==r)
{
segTree[i].nSum=num[l];
return;
}
int mid=(l+r)>>1;
Build(i<<1,l,mid);
Build(i<<1|1,mid+1,r);
segTree[i].nSum=segTree[i<<1].nSum+segTree[i<<1|1].nSum;
}
void Add(int i,int t,int b)
{
segTree[i].nSum+=b;
if(segTree[i].l==t&&segTree[i].r==t) return;
int mid=(segTree[i].l+segTree[i].r)>>1;
if(t<=mid) Add(i<<1,t,b);
else Add(i<<1|1,t,b);
}
int Query(int i,int l,int r)
{
if(l==segTree[i].l&&r==segTree[i].r)
return segTree[i].nSum;
int mid=(segTree[i].l+segTree[i].r)>>1;
if(r<=mid) return Query(i<<1,l,r);
else if(l>mid) return Query(i<<1|1,l,r);
else return Query(i<<1,l,mid)+Query(i<<1|1,mid+1,r);
}
int main()
{
int T;
int iCase=0;
int n,i;
char str[10];
int a,b;
scanf("%d",&T);
while(T--)
{
iCase++;
scanf("%d",&n);
for(i=1;i<=n;i++)
scanf("%d",&num[i]);
Build(1,1,n);
printf("Case %d:\n",iCase);
while(scanf("%s",&str))
{
if(strcmp(str,"End")==0) break;
scanf("%d%d",&a,&b);
if(strcmp(str,"Add")==0) Add(1,a,b);
else if(strcmp(str,"Sub")==0) Add(1,a,-b);
else printf("%d\n",Query(1,a,b));
}
}
return 0;
}
树状数组:
二、离散化
补:
1 左移运算符 <<
有移运算符 >>
x<<1 相当于将x2
x<<1|1 相当于将x2再+1
2