CodeVs 1690 開關燈
阿新 • • 發佈:2017-09-08
struct print include n) 線段樹 lag truct down esp
線段樹區間更新 區間查詢~
好久沒寫過線段樹了 寫挫了好幾次了了了。。
最後瞅了瞅題解才發現自己思路有些問題
#include <cstdio> #include <algorithm> using namespace std; #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 int n,m; struct data{ int l,r,v; bool lazy; }tr[2000001]; void build(int l,int r,int rt){ tr[rt].l=l; tr[rt].r=r; if(l==r) return; int m =(l+r)>>1; build(lson); build(rson); } void pushdown(int rt){ if(!tr[rt].lazy) return; tr[rt<<1].lazy=!tr[rt<<1].lazy; tr[rt<<1|1].lazy=!tr[rt<<1|1].lazy; tr[rt<<1].v=tr[rt<<1].r-tr[rt<<1].l+1-tr[rt<<1].v; tr[rt<<1|1].v=tr[rt<<1|1].r-tr[rt<<1|1].l+1-tr[rt<<1|1].v; tr[rt].lazy=0; } void update(int x,int y,int rt){ pushdown(rt); int l = tr[rt].l; int r = tr[rt].r; if(l==x && r==y){ tr[rt].v=y-x+1-tr[rt].v; if(l!=r) tr[rt].lazy=1; return; } int m =(l+r)>>1; if(y<=m) update(x,y,rt<<1); else if(m<x) update(x,y,rt<<1|1); else { update(x,m,rt<<1); update(m+1,y,rt<<1|1); } tr[rt].v=tr[rt<<1].v+tr[rt<<1|1].v; } int query(int x,int y,int rt){ pushdown(rt); int l = tr[rt].l;int r =tr[rt].r; if(l==x && r==y) return tr[rt].v; int m = (l+r)>>1; if(m>=y) return query(x,y,rt<<1); else if(m<x) return query(x,y,rt<<1|1); else return (query(x,m,rt<<1)+query(m+1,y,rt<<1|1)); } int main(){ scanf("%d%d",&n,&m); build(1,n,1); for(int i=1;i<=m;i++){ int flag,x,y; scanf("%d%d%d",&flag,&x,&y); if(flag) printf("%d\n",query(x,y,1)); else update(x,y,1); } return 0; }
CodeVs 1690 開關燈