CF981E Addition on Segments

我觉得这么 NB 的题为什么只有 2200。

我们考虑用一个 bitset 来维护最大值的可能性。

每一个节点维护一个 bitset \(s\)\(s[i]\)\(1\) 当且经当这个节点对应的区间的最大值可能为 \(i\)

对于每个节点的 \(s\),初值是 \(s[0]=1\)

对于一个操作 \((l,r,k)\),我们显然可以拆成线段树上的 \(log\) 个区间,那么在一个线段树节点上如何维护这个 \(s\) 呢?

我们考虑选或不选这个操作,如果选了,那么这些最大值都会加上 \(k\),在 \(s\) 上表示为 \(s<<k\)

如果不选,\(s\) 还是 \(s\)

所以最终 \(s\) 就等于 \(s|=(s<<k)\)

我们将每个节点的 \(k\) 存起来,然后在最后一次 dfs 的过程中,对每个节点的 \(s\) 都做一次上述操作,然后将信息推到儿子上,最后把叶子节点上的 \(s\) 或起来即可。

时间复杂度 \(O(nlogn+\frac{nqlogn}{w})\)

#include<bits/stdc++.h>
using namespace std;
#define ls p<<1
#define rs p<<1|1
const int N=2e4+8;
inline int read() {
    int s=1,a=0;char c=getchar();
    while(!isdigit(c)) {if(c=='-') s=-s;c=getchar();}
    while(isdigit(c)) a=(a<<3)+(a<<1)+c-'0',c=getchar();
    return s*a;
}
bitset <N> ans,s;
struct sgt {
    vector <int> G[N<<2];
    void change(int l,int r,int p,int s,int t,int k) {
        if(s<=l&&r<=t) {
            G[p].push_back(k);
            return;
        }
        int mid=(l+r)>>1;
        if(s<=mid) change(l,mid,ls,s,t,k);
        if(t>mid) change(mid+1,r,rs,s,t,k);
    }
    void solve(int l,int r,int p,bitset <N> now) {
        bitset <N> thi=now;
        for(auto v:G[p]) thi|=(thi<<v);
        if(l==r) {
            ans|=thi;
            return;
        }
        int mid=(l+r)>>1;
        solve(l,mid,ls,thi);
        solve(mid+1,r,rs,thi);
    }
} wgj;
int n,q;
vector <int> an;
int main() {
    n=read(),q=read();
    for(int i=1; i<=q; i++) {
        int l=read(),r=read(),k=read();
        wgj.change(1,n,1,l,r,k);
    }
    s[0]=1;
    wgj.solve(1,n,1,s);
    for(int i=0; i<=n; i++) {
        if(ans[i]) an.push_back(i);
    }
    printf("%d\n",an.size()-1);
    for(auto v:an) {
        if(v)
        printf("%d ",v);
    }
    return 0;
}
posted @ 2021-10-01 12:20  redproblemdog  阅读(38)  评论(0编辑  收藏  举报