当前位置: 代码迷 >> 综合 >> 【Codeforces Round #275 (Div. 2) D.Interesting Array】线段树
  详细解决方案

【Codeforces Round #275 (Div. 2) D.Interesting Array】线段树

热度:30   发布时间:2023-12-29 02:02:19.0

链接:

CodeForces-483D Interesting Array

题意

让你构造一个数列,满足m种限制条件,每种限制条件是l,r,x,要求构造的序列区间[l,r] 与运算的值结果为x

做法

首先由一个拆位的做法,对于每次询问,拆成30位分别进行构造,对于每个为0的位,区间查询这一位下[l,r]的和是不是等于区间长度,也就是看区间[l,r]与的结果是否为0。对于每个为1的位,区间更新。但是这个做法是两个log的,楼主并有没卡过去。
之后楼主的做法是这样的,首先对于每个查询,我们可以看当前[l,r]区间与的结果,如果这个结果是x的子集(这里的子集是二进制下的01序列),那么说明我们可以通过区间或运算让区间[l,r]的结果变成x,否则一定不行会破坏之前的关系,于是只需要线段树维护区间与运算,区间更新或运算即可。(要注意同一个区间多次询问的情况特判一下)。

代码

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<map>
using namespace std;
typedef pair<int,int> pii;
const int maxn = 1e5+5;
struct T
{
    int l,r,mid;int sum,add;
}tree[maxn<<2];
void push_up(int rt)
{
    tree[rt].sum=tree[rt<<1].sum&tree[rt<<1|1].sum;
}
void push_down(int rt)
{
    if(tree[rt].add!=0){
    int tmp=tree[rt].add;tree[rt<<1].add=tree[rt<<1].add|tmp;tree[rt<<1].sum=tree[rt<<1].sum|tmp;tree[rt<<1|1].add=tree[rt<<1|1].add|tmp;tree[rt<<1|1].sum=tree[rt<<1|1].sum|tmp;tree[rt].add=0;}
}
void build(int rt,int l,int r)
{
    tree[rt].l=l;tree[rt].r=r;tree[rt].sum=0;tree[rt].add=0;if(l==r) return ;int mid=tree[rt].mid=l+r>>1;build(rt<<1,l,mid);build(rt<<1|1,mid+1,r);return ;
}
void update(int rt,int l,int r,int val)
{
    if(tree[rt].l>r||tree[rt].r<l) return ;if(tree[rt].sum==tree[rt].r-tree[rt].l+1) return ;if(tree[rt].l>=l&&tree[rt].r<=r){
    tree[rt].sum=tree[rt].sum|val;tree[rt].add=tree[rt].add|val;return ;}push_down(rt);if(tree[rt].mid>=l) update(rt<<1,l,r,val);if(tree[rt].mid<r) update(rt<<1|1,l,r,val);push_up(rt);return ;
}
int query(int rt,int l,int r)
{
    if(tree[rt].l>r||tree[rt].r<l) return 0;if(tree[rt].l>=l&&tree[rt].r<=r) return tree[rt].sum;push_down(rt);int ans=(1<<30)-1;if(tree[rt].mid>=l) ans=ans&query(rt<<1,l,r);if(tree[rt].mid<r) ans=ans&query(rt<<1|1,l,r);push_up(rt);return ans;
}
int ans[maxn];
map<pii,int> mp;
int main()
{
    int n,m,l,r,q;scanf("%d%d",&n,&m);build(1,1,n);int flag=0;for(int i=1;i<=m;i++){
    scanf("%d%d%d",&l,&r,&q);if(mp.count(pii(l,r))){
    if(mp[pii(l,r)]!=q) flag=1;}mp[pii(l,r)]=q;if(flag==1) continue;int res=query(1,l,r);if((q&res)==res){
    int tmp=q-res;update(1,l,r,tmp);}else flag=1;}if(flag==1) return 0*puts("NO");for(int i=1;i<=n;i++) ans[i]=query(1,i,i);puts("YES");for(int i=1;i<=n;i++) printf("%d ",ans[i]);return 0;
}
  相关解决方案