Sicily 1686 Happy Children's Day(线段树)

//线段树//WA了很多次，最后查出原因居然是建树的时候初始化有最大糖果的下标错了//我只有在叶子节点有初始化，其他点没有，所以错了，正确初始化应该是每个区间最左边的点为maxid//线段树插入和修改的inherit和update函数是关键，直到这道题总算摸索出套路了//建树什么都差不多，关键就看如何识别标记，如何处理区间与区间之间关系，如何回溯传递结果#include<iostream>#define MAX 100005using namespace std;struct seg{int l,r,_max,_maxid,add;}tree[4 * MAX];int index,x,y,add,_maxCandy,N,M;char cmd;void buildTree(int fa,int l,int r){tree[fa].l = l;tree[fa].r = r;tree[fa]._max = tree[fa].add = 0;tree[fa]._maxid = l;if(l == r)return;int mid = (l + r) >> 1;buildTree(2*fa,l,mid);buildTree(2*fa+1,mid+1,r);}void inherit(int fa)//读取当前结点的add,并将这一信息传递给左右子树{tree[2*fa].add += tree[fa].add;tree[2*fa+1].add += tree[fa].add;tree[2*fa]._max += tree[fa].add;tree[2*fa+1]._max += tree[fa].add;tree[fa].add = 0;//注意传递好后就清除ADD的情况，对搜索到的区间，ADD只能增加一次}void update(int fa)//回溯时更新父结点的信息{tree[fa]._max = max(tree[2*fa]._max,tree[2*fa+1]._max);//看左右子树哪个最大tree[fa]._maxid = tree[2*fa]._max >= tree[2*fa+1]._max ? tree[2*fa]._maxid : tree[2*fa+1]._maxid;//哪个最大就记录那个的ID,优先记录左子树}void insert(int fa,int l,int r,int add){if(tree[fa].l == l && tree[fa].r == r){tree[fa].add += add;tree[fa]._max += add;return;}inherit(fa);int mid = (tree[fa].l + tree[fa].r) >> 1;if(r <= mid)insert(2*fa,l,r,add);else if(l > mid)insert(2*fa+1,l,r,add);else {insert(2*fa,l,mid,add);insert(2*fa+1,mid+1,r,add);}update(fa);}void search(int fa,int l,int r){if(tree[fa].l == l && tree[fa].r == r){if(tree[fa]._max == _maxCandy)index = (tree[fa]._maxid,index);if(tree[fa]._max > _maxCandy){_maxCandy = tree[fa]._max;index = tree[fa]._maxid;}return;}inherit(fa);int mid = (tree[fa].l + tree[fa].r) >> 1;if(r <= mid)search(2*fa,l,r);else if(l > mid)search(2*fa+1,l,r);else{search(2*fa,l,mid);search(2*fa+1,mid+1,r);}update(fa);}int main(){while(scanf("%d%d",&N,&M) && N){buildTree(1,1,N);while(M--){scanf("/n%c",&cmd);if(cmd == 'I'){scanf("%d%d%d",&x,&y,&add);insert(1,x,y,add);}else {_maxCandy = 0;//最大糖果数量index = N;//拥有最大糖果堆的下表scanf("%d%d",&x,&y);search(1,x,y);//搜索改区间printf("%d/n",_maxCandy);if(_maxCandy != 0)insert(1,index,index,-_maxCandy);//利用插入的方法，插入个负的，把糖拿出来，如果最大的是0那就不用修改了，省点时间}}}return 0;}