2019牛客暑期多校训练营(第四场)B xor (线性基+线段树)

https://ac.nowcoder.com/acm/contest/884/B
题意:对于一个集合,若该集合的某个子集的异或和等于 x x x,则称该集合可以“表示” x x x,给定个 n n n个集合和 m m m次询问,每次询问给定 l , r , x l,r,x l,r,x,表示询问对于任意 i ∈ [ l , r ] i \in [l,r] i[l,r] i i i个集合中的数能否表示 x x x 1 ≤ n , m ≤ 50000 1≤n,m≤50000 1n,m50000

线性基套线段树,线段树每个节点是一个线性基,父结点是子节点线性基的交,线性基板子参考该博客(主要是线性基的交怎么写)https://blog.csdn.net/sssSSSay/article/details/97615912
判断 x x x是否能在该线性基中表达和插入操作一样,只是不进行插入而已。

#include 
#include 
#include 
#include 
#include 
using namespace std;

struct L_B{
    long long d[61],p[61];
    int cnt;
    L_B()
    {
        memset(d,0,sizeof(d));
        memset(p,0,sizeof(p));
        cnt=0;
    }
    bool insert(long long val)
    {
        for (int i=60;i>=0;i--)
            if (val&(1LL<<i))
            {
                if (!d[i])
                {
                    d[i]=val;
                    break;
                }
                val^=d[i];
            }
        return val>0;
    }
    void rebuild()
    {
        for (int i=60;i>=0;i--)
            for (int j=i-1;j>=0;j--)
                if (d[i]&(1LL<<j))
                    d[i]^=d[j];
        for (int i=0;i<=60;i++)
            if (d[i])
                p[cnt++]=d[i];
    }
    bool query(long long val)
    {
        bool flag=1;
        for (int i=60;i>=0;i--)
            if (val&(1LL<<i))
            {
                if (!d[i])
                {
                    flag=0;
                    break;
                }
                val^=d[i];
            }
        return flag;
    }
}a[50050];
L_B Merge(L_B A, L_B B) {
	L_B All, C, D;
	for(int i = 32; i >= 0; i--) {
		All.d[i] = A.d[i];
		D.d[i] = 1ll << i;
	}
	for(int i = 32; i >= 0; i--)
		if(B.d[i]) {
			long long v = B.d[i], k = 0;
			bool can = true;
			for(int j = 32; j >= 0; j--)
				if(v & (1ll << j)) {
					if(All.d[j]) {
						v ^= All.d[j];
						k ^= D.d[j];
					}
					else {
						can = false;
						All.d[j] = v;
						D.d[j] = k;
						break;
					}
				}
			if(can) {
				long long v = 0;
				for(int j = 32; j >= 0; j--)
					if(k & (1ll << j))
						v ^= A.d[j];
				C.insert(v);
			}
		}
	C.rebuild();
	return C;
}
struct Seg{
    L_B v;
    Seg *ls,*rs;
};
//建树
void build(Seg* &p,int l,int r)
{
    p=new Seg();
    p->v=L_B();
    if(l==r)
    {
        p->v=a[l];
        return ;
    }
    int mid=(l+r)/2;
    build(p->ls,l,mid);
    build(p->rs,mid+1,r);
    p->v=Merge(p->ls->v,p->rs->v);
}
//区间查询[L,R]
bool query(Seg* &p,long long x,int l,int r,int L,int R)
{
    if(l>R||r<L) return true;
    else if(L<=l&&r<=R) return p->v.query(x);
    else
    {
        int mid=(l+r)/2;
        return query(p->ls,x,l,mid,L,R)&&query(p->rs,x,mid+1,r,L,R);
    }
}
//内存回收
void destruct(Seg* &p) {
    if(p == NULL)  return;
    destruct(p -> ls);
    destruct(p -> rs);
    delete p;
    p = NULL;
}

int main()
{
    Seg* root;
    int n,m,sz,l,r;
    long long x;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&sz);
        for(int j=1;j<=sz;j++)
        {
            scanf("%lld",&x);
            a[i].insert(x);
        }
    }
    build(root,1,n);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%lld",&l,&r,&x);
        if(query(root,x,1,n,l,r))
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}

你可能感兴趣的:(线性基)