COJ 0712 Can you answer these queries II

------------------------------------------------------------------------------------

线段树的区间操作：修改区间内的值，反转区间内的值，区间查询；

------------------------------------------------------------------------------------

Description

维护一个只有0和1的整数序列，支持以下操作：
1 x y v : 将区间[x,y]之间的所有整数都变为v（v为0或1）；
2 x y : 将区间[x,y]之间所有的1变为0，所有的0变为1；
3 x y : 查询区间[x,y]内的1的个数。

Input

输入文件包含多组测试数据。
对于每组测试数据，第一行为一个整数N(1<=N<=10^5)，表示这个序列一共有N个整数。接下来一行有N个值为0或1的整数，分别描述了序列中的N个整数。再接下来一行有一个整数M(1<=M<=10^5)，表示一共有M个操作。

Output

对于每个查询均用一行输出查询的结果。

Sample Input

4 0 0 1 0 6 3 1 4 2 1 4 3 1 4 1 1 2 0 1 3 3 1 3 1 4

Sample Output

1 3 2

------------------------------------------------------------------------------------

# include <stdio.h>



# define N 100015

# define NON 0x7fffffff



int n, m, num[N];

int ones[4 * N], set[4 * N], rev[4 * N];



void update(int r)

{

    ones[r] = ones[r << 1] + ones[(r << 1) | 1];

}



void pushdown(int r, int x, int y)

{

    int mid = (x+y) >> 1, ls = r << 1, rs = (r << 1) | 1;



    if (set[r] != NON)

    {

        rev[ls] = rev[rs] = 0;

        set[ls] = set[rs] = set[r], set[r] = NON;

        ones[ls] = (mid-x+1) * set[ls];

        ones[rs] = (y-mid) * set[rs];

    }

    if (rev[r])

    {

        rev[ls] ^= 1;

        rev[rs] ^= 1;

        ones[ls] = mid-x+1 - ones[ls];

        ones[rs] = y - mid - ones[rs];

        rev[r] = 0;    

    }

}



void build(int r, int x, int y)

{

    set[r] = NON, rev[r] = 0;

    if (x == y)

    {

        ones[r] = num[x];

        return ;

    }

    int mid = (x + y) >> 1, ls = r << 1, rs = (r << 1) | 1;

    build(ls, x, mid);

    build(rs, mid+1, y);

    update(r);

}



void change(int r, int x, int y, int s, int t, int val)

{

    if (s<=x && y<=t)

    {

        ones[r] = (y-x+1) * val;

        set[r] = val;

        rev[r] = 0;

        return ;

    }

    pushdown(r, x, y);

    int mid = (x+y) >> 1, ls = r << 1, rs = (r << 1) | 1;

    if (s <= mid) change(ls, x, mid, s, t, val);

    if (mid+1 <= t) change(rs, mid+1, y, s, t, val);

    update(r);

}



void flip(int r, int x, int y, int s, int t)

{

    if (s<=x && y<=t)

    {

        rev[r] ^= 1;

        ones[r] = y-x+1 - ones[r];

        return ;

    }

    pushdown(r, x, y);

    int mid = (x+y) >> 1, ls = r << 1, rs = (r << 1) | 1;

    if (s <= mid) flip(ls, x, mid, s, t);

    if (mid+1 <= t) flip(rs, mid+1, y, s, t);

    update(r);

}



void query(int r, int x, int y, int s, int t, int &ans)

{

    if (s<=x && y<=t)

    {

        ans += ones[r];

        return ;

    }

    pushdown(r, x, y);

    int mid = (x+y) >> 1, ls = r << 1, rs = (r << 1) | 1;

    if (s <= mid) query(ls, x, mid, s, t, ans);

    if (mid+1 <= t) query(rs, mid+1, y, s, t, ans);

}



void init(void)

{

    for (int i = 1; i <= n; ++i)

    {

        scanf("%d", &num[i]);

    }

    scanf("%d", &m);

    build(1, 1, n);

}



void solve(void)

{

    int id, val, s, t, ans;



    for (int i = 1; i <= m; ++i)

    {

        scanf("%d", &id);

        switch(id)

        {

        case 1: {scanf("%d%d%d", &s, &t, &val); change(1, 1, n, s, t, val); break;}

        case 2: {scanf("%d%d", &s, &t); flip(1, 1, n, s, t); break;}

        case 3: {scanf("%d%d", &s, &t); ans = 0; query(1, 1, n, s, t, ans); printf("%d\n", ans); break;}

        }

    }



    return ;

}



int main()

{

    while (~scanf("%d", &n))

    {

        init();

        solve();

    }



    return 0;

}

------------------------------------------------------------------------------------