CF981E Addition on Segments 线段树分治 + bitset优化

最新推荐文章于 2021-04-06 11:41:01 发布

YB Lin

最新推荐文章于 2021-04-06 11:41:01 发布

阅读量150

点赞数

分类专栏：数据结构线段树分治

本文链接：https://blog.csdn.net/m0_51068403/article/details/114969799

版权

数据结构同时被 2 个专栏收录

113 篇文章 4 订阅

订阅专栏

线段树分治

2 篇文章 0 订阅

订阅专栏

传送门

文章目录

题意：
思路：

题意：

在这里插入图片描述

思路：

我们考虑如果我们选择的区间都包含某个位置，那么这个位置一定是最大值。那么对于每个位置，我们枚举包含其的区间，让后每次加 $x$ 都用 $b i t s e t$ 来维护，即 $b i t ∣ = (b i t < < x)$ ，最后 $1 - n$ 中 $1$ 的位置即为答案。但是这样写的话复杂度是 $\frac {n^2q}{w}$ ，显然不能接受。考虑如何优化掉一层 $n$ 呢？可以发现对于每个位置，我们只需要考虑包含这个位置的区间，所以我们用线段树分治将每个区间分割成 $l o g n$ 段，从根开始遍历，如果到了当前区间就把当前区间的贡献加上，生成一个新集合(原来的数以及原来的数+新的数)，让后到一个点的时候就把答案更新一下。这样就可以保证对于每个点我们都能加上包含这个点的区间贡献。
复杂度 $O(\frac{nqlogn}{w})$

//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<bitset>
#include<ctime>
#include<cstdlib>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std;

//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int N=10010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n,m;
bitset<N>s;
struct Node
{
    int l,r;
    vector<int>v;
}tr[N<<2];

void build(int u,int l,int r)
{
    tr[u]={l,r};
    if(l==r) return;
    build(L,l,Mid); build(R,Mid+1,r);
}

void insert(int u,int l,int r,int x)
{
    if(tr[u].l>=l&&tr[u].r<=r)
    {
        tr[u].v.pb(x);
        return;
    }
    if(l<=Mid) insert(L,l,r,x);
    if(r>Mid) insert(R,l,r,x);
}

void dfs(bitset<N>pre,int u,int l,int r)
{
    bitset<N>now=pre;
    for(auto x:tr[u].v) now|=(now<<x);
    int mid=l+r>>1;
    if(l==r) { s|=now; return; }
    else dfs(now,L,l,mid),dfs(now,R,mid+1,r);
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);

    scanf("%d%d",&n,&m);
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        int l,r,x; scanf("%d%d%d",&l,&r,&x);
        insert(1,l,r,x);
    }
    s[0]=1;
    dfs(s,1,1,n);
    int ans=0;
    for(int i=1;i<=n;i++) if(s[i]) ans++;
    cout<<ans<<endl;
    for(int i=1;i<=n;i++) if(s[i]) printf("%d ",i);



	return 0;
}
/*

*/