Pytorch学习(2) —— 快速模型

好久不写题解了，今天写了一个线段树果然一点手感都没有了。

不过是个很好的题目，两种标记是互相影响的，注意到什么标记的更新会使得另外的标记一并更新
题目和解法见注释

/**
 * UVA 11402
 * 四种操作
 * 注意两种标记会相互影响的时候的处理
 */
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define lson rt<<1
#define rson rt<<1|1

using namespace std;
const int MAXN = 1100080; 
struct Segment{
    int l,r;
    int sum,len;
    int rev,tag;// tag = 1 表示区间中都是１，tag = 0 表示区间中都是0，tag = -1 表示区间中01不一定
}tree[MAXN << 2];
int n,a[MAXN];

inline void pushup(int rt){
    tree[rt].sum = tree[lson].sum + tree[rson].sum;
    tree[rt].tag = -1;
    tree[rt].rev = 0;
}

/**
 * 这里，关于pushdown函数，传递的到底应该是标记
 * 还是说可以传递数据
 */
inline void pushdown(int rt){
    if(tree[rt].tag == 1){
        tree[lson].tag = tree[rson].tag = 1;
        tree[lson].sum = tree[lson].len;
        tree[rson].sum = tree[rson].len;
        tree[lson].rev = tree[rson].rev = 0;
        tree[rt].tag = -1;
    }
    else if(tree[rt].tag == 0){
        tree[lson].tag = tree[rson].tag = 0;
        tree[lson].sum = tree[rson].sum = 0;
        tree[lson].rev = tree[rson].rev = 0;
        tree[rt].tag = -1;
    }

    if(tree[rt].rev == 1){
        tree[lson].rev = 1 - tree[lson].rev;
        tree[rson].rev = 1 - tree[rson].rev;
        // tree[rt].sum = tree[rt].len - tree[rt].sum;
        tree[lson].sum = tree[lson].len - tree[lson].sum;       
        tree[rson].sum = tree[rson].len - tree[rson].sum;
        tree[rt].rev = 0;
    }
}

inline void build(int rt,int l,int r){
    tree[rt].l = l;
    tree[rt].r = r;
    tree[rt].len = r - l + 1; 
    if(l == r){
        tree[rt].rev = 0;
        tree[rt].tag = -1;
        tree[rt].sum = a[l];
        return ;
    }
    int mid = (l + r) >> 1;

    build(lson,l,mid);
    build(rson,mid+1,r);

    pushup(rt);
}
inline void update_rev(int rt,int l,int r){
    /*
     *　这里，当整个节点都在更改的位置中
     *  所以整段可以直接更新、返回
     */
    if(l <= tree[rt].l && tree[rt].r <= r){
        tree[rt].rev = 1 - tree[rt].rev;
        tree[rt].sum = tree[rt].len - tree[rt].sum;
        // tree[rt].sum = tree[rt].len - tree[rt].sum;
        return ;
    }
    int mid = (tree[rt].l + tree[rt].r) >> 1;
    pushdown(rt);

    if(l <= mid) update_rev(lson,l,r);
    if(r > mid) update_rev(rson,l,r);

    pushup(rt);
}


inline void update(int rt,int l,int r,int data){
    /*
     *　这里，当整个节点都在更改的位置中
     *  所以整段可以直接更新、返回
     */
    if(l <= tree[rt].l && tree[rt].r <= r){
        tree[rt].tag = data;
        tree[rt].sum = tree[rt].len * data;
        tree[rt].rev = 0;
        return ;
    }
    int mid = (tree[rt].l + tree[rt].r) >> 1;
    pushdown(rt);

    if(l <= mid) update(lson,l,r,data);
    if(r > mid) update(rson,l,r,data);

    pushup(rt);
}

inline int query(int rt,int l,int r){
    if(l <= tree[rt].l && tree[rt].r <= r){
        return tree[rt].sum;
    }
    pushdown(rt);
    int mid = (tree[rt].l + tree[rt].r) >> 1;

    if(r <= mid) return query(lson,l,r);
    else if(l >= mid+1) return query(rson,l,r);
    else return query(lson,l,r) + query(rson,l,r);
}

char ch[105];

int main(){
    int T,nc = 1;
    scanf("%d",&T);
    for(int cas = 1;cas <= T;cas ++){
        int tmp,tt = 0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            memset(ch,0,sizeof ch);
            scanf("%d\n%s",&tmp,ch);
            int chlen = strlen(ch);
            for(int t = 0;t < tmp;t++){
                for(int j = 0;j < chlen;j++){
                    a[++tt] = ch[j] - '0';
                }
            }
        }

        build(1,1,tt);

        char op[2];
        int l,r,qm,qq = 0;
        scanf("%d",&qm);
        printf("Case %d:\n",cas);
        for(int Q = 1;Q <= qm;Q ++){
            scanf("%s%d%d",op,&l,&r);
            l ++,r ++;
            if(op[0] == 'F'){
                // F 置为１
                update(1,l,r,1);                
            }
            else if(op[0] == 'E'){
                // E　置为０
                update(1,l,r,0);
            }
            else if(op[0] == 'I'){
                // I 反
                update_rev(1,l,r);  
            }
            else if(op[0] == 'S') {
                // ask
                int ans = query(1,l,r);
                printf("Q%d: %d\n",++qq,ans);
            }
        }
    }

    return 0;
}