牛客练习赛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;
}
复杂度显然只有