欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

牛客练习赛49-筱玛爱线段树 思维

程序员文章站 2022-07-12 17:39:25
...

太久没做题了,看着有牛客就打了一场,果然自闭…

原题链接:https://ac.nowcoder.com/acm/contest/946/D

题目大意也很明确,给定一个长度为n,初始全为0的树组。给定以下两种操作:

  • 1 l r :将树组中[l,r]中的数都加1
  • 2 l r : 执行操作编号在[l,r]内的所有操作各一次,题目保证r小于当前操作的编号。

思维退化太快。开始老是想着建一个可持久化的,或者怎瞎搞,发现复杂度都降不下来。后来绝处逢生,想到了将所有的操作倒着做,记录每一个操作被后面的操作总共引用了多少次,完美解决。复杂度也能过。
最绝望的来了,由于太久没写题,导致树状树组的意义都忘了,一遍吃饭一遍调树状树组,调了一个小时左右,终于调过了样例。然后马上交了一发,果不其然的WA了。此时距离比赛结束还有十几分钟,最后果不其然的凉了。

最后WA的地方对于当时的我来说可能确实有点难看出来(但是样例却神奇的过了),第二天对着别人的代码debug才发现哪里出了错(在下方的代码中标注了)…

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
const int mod=1e9+7;
int tr[N],p[N];
inline int lowbit(int x){return x&(-x);}
void add(int *tr,int x,int d){
    for(;x<N;x+=lowbit(x))
        tr[x]=(tr[x]+d)%mod;
}
int query(int *tr,int x){
    int res=0;
    for(;x;x-=lowbit(x))
        res=(res+tr[x])%mod;
    return res;
}
int opt[N],l[N],r[N];
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    add(tr,1,1);
    add(tr,m+1,mod-1);
    // for(int i=1;i<=m;i++){
    //    printf("%d%c",query(tr,i)%mod,i==m?'\n':' ');
    // }
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&opt[i],&l[i],&r[i]);
    }
    for(int i=m;i>=1;i--){
        int tmp=query(tr,i);
        if(opt[i]==1){
            add(p,l[i],tmp);
            add(p,r[i]+1,mod-tmp);
        }
        else{
            add(tr,l[i],tmp);//WA的就是这个地方
            add(tr,r[i]+1,mod-tmp);
        }
       	/*
       	开始的写法:
       	if(opt[i]==1){
            add(p,l[i],query(tr,i));
            add(p,r[i]+1,mod-query(tr,i));
        }
        else{
            add(tr,l[i],query(tr,i));
            add(tr,r[i]+1,mod-query(tr,i));
        }
        这样显然是不行的,因为前面在更新的时候会改变当前tr[i]的值
       	*/
    }
    for(int i=1;i<=n;i++){
        printf("%d%c",query(p,i),i==n?'\n':' ');
    }
    return 0;
}

复杂度显然只有O(NlogN)O(NlogN)