作诗 --- 分块(区间众数)

传送门洛谷P4135


题目描述

神犇SJY虐完HEOI之后给傻×LYD出了一题:

SHY是T国的公主,平时的一大爱好是作诗。

由于时间紧迫,SHY作完诗之后还要虐OI,于是SHY找来一篇长度为N的文章,阅读M次,每次只阅读其中连续的一段[l,r],从这一段中选出一些汉字构成诗。因为SHY喜欢对偶,所以SHY规定最后选出的每个汉字都必须在[l,r]里出现了正偶数次。而且SHY认为选出的汉字的种类数(两个一样的汉字称为同一种)越多越好(为了拿到更多的素材!)。于是SHY请LYD安排选法。

LYD这种傻×当然不会了,于是向你请教……

问题简述:N个数,M组询问,每次问[l,r]中有多少个数出现正偶数次。


分析

  与求区间众数差不多吧([Violet]蒲公英)
  1.先预处理处块内每个种类的个数,再前缀和处理一下
  2.求出块与块之间的ans
  3.对于询问:只要处理非完整块中的种类即可(完整块中的提前加上,到时候判断即可)


代码写得太丑,常数有点大,望大佬指教。

代码

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>

#define IL inline
#define open(s) freopen(s".in","r",stdin); freopen(s".out","w",stdout);
#define close fclose(stdin); fclose(stdout);

using namespace std;

IL int read()
{
    char c = getchar();
    int sum = 0 ,k = 1;
    for(;'0' > c || c > '9'; c = getchar())
        if(c == '-') k = -1;
    for(;'0' <= c && c <= '9'; c = getchar()) sum = sum * 10 + c - '0';
    return sum * k;
}

const int maxn = 100000 + 5, maxT = 320;

int n, k, m, T;
int block;
int num[maxn];
int cnt[maxT][maxn];
int Ans[maxT][maxT];
int use[maxn];

struct Block
{
    int l, r;
}ip[maxT];

IL void add(int w, int &val)
{
    if(w && !(w & 1)) ++val;
    if(w > 1 && (w & 1)) --val;
}

IL void pre_work()
{
    for(int i = 2; i < T; ++i)
    for(int j = 1; j <= k; ++j)
        cnt[i][j] += cnt[i - 1][j]; // 前缀和

    for(int i = 1, s, an; i < T; ++i)
    {
        an = 0;
        for(int j = i; j < T; ++j)
        {
            for(int l = ip[j].l; l <= ip[j].r; ++l) //枚举这个块上的种类就够了, 不用枚举全部种类 
                add(++use[num[l]], an);
            Ans[i][j] = an;
        }
        for(int j = ip[i].l; j <= n; ++j)
            if(use[num[j]]) use[num[j]] = 0;
    }
}

IL int query(int x, int y)
{
    int l = x / block + 1, r = y / block + 1, ans;

    if(l + 1 >= r)
    {
        ans = 0;
        for(int i = x, s; i <= y; ++i)
            add(++use[num[i]], ans);

        for(int i = x; i <= y; ++i) 
            if(use[num[i]]) use[num[i]] = 0;
    }else
    {
        ans = Ans[l + 1][r - 1];
        for(int i = x; i <= ip[l].r; ++i)
        {
            if(!use[num[i]]) use[num[i]] = cnt[r - 1][num[i]] - cnt[l][num[i]];
            add(++use[num[i]], ans);
        }

        for(int i = ip[r].l; i <= y; ++i)
        {
            if(!use[num[i]]) use[num[i]] = cnt[r - 1][num[i]] - cnt[l][num[i]];
            add(++use[num[i]], ans);
        }

        for(int i = x; i <= ip[l].r; ++i)
            if(use[num[i]]) use[num[i]] = 0;

        for(int i = ip[r].l; i <= y; ++i)
            if(use[num[i]]) use[num[i]] = 0;
    }
    return ans;
}

int main()
{
    open("4135")

    n = read(); k = read(); m = read(); block = sqrt( (double)n / log((double)n) * log(2)); T = 1;
    for(int i = 1; i <= n; ++i)
    {
        num[i] = read();
        ++cnt[T][num[i]];
        if(!(i % block) || i == n)
        {
            ip[T].r = i;
            ip[++T].l = i + 1;
        }
    }

    pre_work();

    for(int pre = 0, x, y; m; --m)
    {
        x = (read() + pre) % n + 1;
        y = (read() + pre) % n + 1;

        if(x > y) swap(x, y);

        printf("%d\n", pre = query(x, y));

    }

    close
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_27121257/article/details/81091618