bzoj 4722: 由乃 线段树+倍增+搜索

题意

给一个长为n的序列a，每个数在0到v - 1之间，有m次操作。
操作1：每次询问一个区间中是否可以选出两个下标的集合X，Y，满足：
1.X和Y没有交集
2.设集合X中有一个元素是i，则其对集合X的贡献是a[i] + 1，要求集合X的元素的总贡献和集合Y的元素的总贡献相等如果可以选出这两个集合，输出 Yuno否则输出 Yuki
操作2：修改一个区间l,r之间的数，使得所有l <= i <= r，a[i] = a[i] * a[i] * a[i] % v ，即区间立方
n , m <= 100000 , v <= 1000

分析

乍一看根本不会做。
网上说只要序列长度大于13则答案必然为Yuno。证明：因为子集数量有$2^{len}$种，而权值则最多只有$len*1000$种，解方程可得$len=13$。
有了这个结论这题就好做了。
对于一个长度不超过13的序列，显然每个数的系数只有0,1,-1三种，我们只要通过折半搜索来判断是否有贡献相同的子集即可。
至于区间开三次方，我们可以通过线段树打标记和倍增来实现。设bz[i,j]表示$i^{3^{2^j}}$,预处理后便可以在O(log)的复杂度内得到一个数。
复杂度$O(n*3^7)$，有点虚但实际上跑得贼快。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=100005;
const int M=15000;

int n,m,v,bz[1005][20],size[N*4],a[N],tot,t[N],f[20];
bool vis[N],flag;

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void prework()
{
    for (int i=0;i<v;i++) bz[i][0]=i*i*i%v;
    for (int j=1;j<=16;j++)
        for (int i=0;i<v;i++)
            bz[i][j]=bz[bz[i][j-1]][j-1];
}

void ins(int d,int l,int r,int x,int y)
{
    if (x>y) return;
    if (l==x&&r==y)
    {
        size[d]++;return;
    }
    int mid=(l+r)/2;
    ins(d*2,l,mid,x,min(y,mid));
    ins(d*2+1,mid+1,r,max(x,mid+1),y);
}

int query(int d,int l,int r,int x)
{
    if (l==r) return size[d];
    int mid=(l+r)/2;
    if (x<=mid) return query(d*2,l,mid,x)+size[d];
    else return query(d*2+1,mid+1,r,x)+size[d];
}

int get(int x)
{
    int y=query(1,1,n,x),ans=a[x];
    for (int i=16;i>=0;i--)
        if (y&(1<<i)) ans=bz[ans][i];
    return ans;
}

void dfs1(int x,int y,int z,int n)
{
    if (x>n)
    {
        vis[y+M]=1;t[++tot]=y;
        if (!y&&z) flag=1;
        return;
    }
    dfs1(x+1,y,z,n);
    if (flag) return;
    dfs1(x+1,y+f[x]+1,z+1,n);
    if (flag) return;
    dfs1(x+1,y-f[x]-1,z+1,n);
}

void dfs2(int x,int y,int z,int n)
{
    if (x>n)
    {
        if (vis[-y+M]&&y!=0||!y&&z) flag=1;
        return;
    }
    dfs2(x+1,y,z,n);
    if (flag) return;
    dfs2(x+1,y+f[x]+1,z+1,n);
    if (flag) return;
    dfs2(x+1,y-f[x]-1,z+1,n);
}

bool solve(int n)
{
    tot=flag=0;
    dfs1(1,0,0,n/2);
    dfs2(n/2+1,0,0,n);
    for (int i=1;i<=tot;i++) vis[t[i]+M]=0;
    return flag;
}

int main()
{
    n=read();m=read();v=read();
    for (int i=1;i<=n;i++) a[i]=read();
    prework();
    while (m--)
    {
        int op=read(),l=read(),r=read();
        if (op==2) ins(1,1,n,l,r);
        else
        {
            if (r-l+1>13)
            {
                puts("Yuno");
                continue;
            }
            for (int j=l;j<=r;j++) f[j-l+1]=get(j);
            if (solve(r-l+1)) puts("Yuno");
            else puts("Yuki");
        }
    }
    return 0;
}