P1494 [国家集训队] 莫队

最新推荐文章于 2024-04-09 19:12:31 发布

SHOHOKUKU

最新推荐文章于 2024-04-09 19:12:31 发布

阅读量112

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/neweryyy/article/details/114055594

版权

数据结构专栏收录该内容

203 篇文章 1 订阅

订阅专栏

题意

传送门 P1494 [国家集训队]小Z的袜子

题解

莫队算法基本思想是对询问进行分块，设询问数为 $M$ 。先将询问离线下来，把这些询问按照左端点递增排序，然后分成 $\sqrt M$ 块，每块内部再按照右端点排序。一种大概率可以进行优化的方法是将块按照奇偶性分别递增、递减排序，使块与块过渡时，查询的右界变化更加平滑。

将相邻块间的最大左端点之差看做一个差分数组 $d$ ，由于询问按照左端点递增排序，则差分数组的前缀和为 $N$ ，最坏情况下块 $i$ 内各相邻询问间左端点之差为 $d [i]$ ，则左端点总的变化范围为 $N\sqrt M$ ；各块内右端点递增，变化范围为 $O (N)$ ，则右端点总的变化范围为 $O(N\sqrt M)$ 。顺序扫描询问，每次以上一次的询问为基础，按照查询区间处理左右界的变化，总时间复杂度 $O(N\sqrt M)$ 。

假设颜色为 $c$ 的袜子数量为 $cnt_c$ ，它对答案分子，即抽到同色袜子对的贡献为 $cnt_c\times (cnt_c-1)/2$ 。答案为 $\sum cnt_c\times (cnt_c-1)/2/\text{C}_{r-l+1}^{2}=\sum [cnt_c\times (cnt_c-1)]/[(r-l+1)\times (r-l)]$

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 50005, maxq = 255;
struct node
{
    int l, r, k;
} Q[maxn];
int N, M, fg, tot, C[maxn], cnt[maxn], L[maxq], R[maxq];
ll tmp, tx, ty, rx[maxn], ry[maxn];

bool cmp_l(const node &a, const node &b) { return a.l < b.l; }

bool cmp_r(const node &a, const node &b) { return fg ? a.r < b.r : a.r > b.r; }

ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }

inline void add(int i, int x)
{
    int &t = cnt[C[i]];
    tx -= (ll)t * (t - 1);
    t += x;
    tx += (ll)t * (t - 1);
}

int main()
{
    scanf("%d%d", &N, &M);
    for (int i = 1; i <= N; ++i)
        scanf("%d", C + i);
    for (int i = 1; i <= M; ++i)
        scanf("%d%d", &Q[i].l, &Q[i].r), Q[i].k = i;
    sort(Q + 1, Q + M + 1, cmp_l);
    int t = sqrt(M);
    t = t ? t : M;
    for (int i = 1; i <= M; i += t)
    {
        L[++tot] = i, R[tot] = min(i + t, M + 1);
        fg = i & 1;
        sort(Q + L[tot], Q + R[tot], cmp_r);
    }
    int l = Q[1].l, r = Q[1].l - 1;
    for (int i = 1, ql, qr, k; i <= M; ++i)
    {
        ql = Q[i].l, qr = Q[i].r, k = Q[i].k;
        while (r < qr)
            add(++r, 1);
        while (r > qr)
            add(r--, -1);
        while (l < ql)
            add(l++, -1);
        while (l > ql)
            add(--l, 1);
        ty = (ll)(qr - ql) * (qr - ql + 1);
        if (!tx)
            rx[k] = 0, ry[k] = 1;
        else
            tmp = gcd(tx, ty), rx[k] = tx / tmp, ry[k] = ty / tmp;
    }
    for (int i = 1; i <= M; ++i)
        printf("%lld/%lld\n", rx[i], ry[i]);
    return 0;
}