[BZOJ 2054]疯狂的馒头

 

这题一看是区间染色，直接上线段树。数据范围N<= 10⁶,M <= 10⁷,如果跑线段树复杂度为O(MlogN)，但是时间限制10s，所以复杂度能过。

具体操作就是倒着算区间，因为每个馒头最后的颜色是最后一次染上的颜色，如果我们倒着染色，那么被染色过的点就是最后的颜色，之后再染这个点就没用了。所以直接上代码：

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define N 1000100
#define ls (x << 1)
#define rs (x << 1 | 1)
#define mid ((l + r) >> 1)
using namespace std;
int tree[4 * N];
void push_up(int x)
{
    tree[x] = tree[ls] && tree[rs];//除了叶子节点之外的点只需要保存是否染过色就行
    return;
}
void modify(int x,int nl,int nr,int l,int r,int k)
{
    if(tree[x]) return;
    if(l == r) 
    {
        tree[x] = k;//叶子节点直接保存颜色
        return;
    }
    if(nl <= mid) modify(ls,nl,nr,l,mid,k);
    if(nr > mid) modify(rs,nl,nr,mid + 1,r,k);
    push_up(x);
    return;
}
void print(int x,int l,int r)
{
    if(l == r) 
    {
        printf("%d\n",tree[x]);//输出叶子节点
        return;
    }
    print(ls,l,mid);
    print(rs,mid + 1,r);
}
int p,q,n,m;
int main()
{
    scanf("%d %d %d %d",&n,&m,&p,&q);
    for(int i = m;i >= 1;i--)
    {
        int l = (int)((long long)(i * p + q) % n + 1ll);//这里强制类型转换防止在做乘法的时候爆int
        int r = (int)((long long)(i * q + p) % n + 1ll);
        if(l > r) swap(l,r);
        modify(1,l,r,1,n,i);
    }
    print(1,1,n);
}

 

转载于:https://www.cnblogs.com/lijilai-oi/p/10945070.html