BZOJ-3781 小B的询问

最新推荐文章于 2019-07-24 21:33:00 发布

weixin_30502965

最新推荐文章于 2019-07-24 21:33:00 发布

阅读量55

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/NanoApe/p/4324588.html

本文探讨了BZOJ2038问题的简化版本及其解决方法，利用分块莫队算法进行优化。通过实例分析，详细解释了算法的具体步骤和实现细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

BZOJ2038的弱化版。。。

同样分块莫队。

#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <fstream>
#include <iostream>
#include <queue>

#define rep(i, l, r) for(int i = l; i <= r; i++)
#define down(i, l, r) for(int i = l; i >= r; i--)
#define N 56789
#define MAX 1<<30
#define ll long long

using namespace std;

struct node{int l, r, id; ll a;} q[N];
int n, m, k, c[N], pos[N];
ll ans, s[N];
ll sqr(ll x) { return x*x; }
bool cmp(node a, node b)
{
	if (pos[a.l] == pos[b.l]) return a.r < b.r; else return a.l < b.l;
}
bool cmp_id(node a, node b) { return a.id < b.id; }

void update(int p, int x)
{
	ans -= sqr(s[c[p]]);
	s[c[p]] += x;
	ans += sqr(s[c[p]]);
}

int main()
{
	scanf("%d%d%d", &n, &m ,&k);
	rep(i, 1, n) scanf("%d", &c[i]);
	int block = int(sqrt(n));
	rep(i, 1, n) pos[i] = (i-1)/block+1;
	rep(i, 1, m) scanf("%d%d", &q[i].l, &q[i].r);
	rep(i, 1, m) q[i].id = i;
	sort(q+1, q+1+m, cmp);
	int l = 1, r = 0;
	rep(i, 1, m)
	{
		while (l < q[i].l) update(l++, -1);
		while (l > q[i].l) update(--l, 1);
		while (r < q[i].r) update(++r, 1);
		while (r > q[i].r) update(r--, -1);
		q[i].a = ans;
	}
	sort(q+1, q+1+m, cmp_id);
	rep(i, 1, m) printf("%lld\n", q[i].a);
	return 0;
}

转载于:https://www.cnblogs.com/NanoApe/p/4324588.html