首页 > 其他 > 详细

CodeForces 86D(Yandex.Algorithm 2011 Round 2)

时间:2014-07-16 18:16:33      阅读:420      评论:0      收藏:0      [点我收藏+]

思路:莫队算法,离线操作,将所有询问的左端点进行分块(分成sqrt(n) 块每块sqrt(n)个),用左端点的块号进行排序小的在前,块号相等的,右端点小的在前面。 这样要是两个相邻的查询在同一块内左端点每次最多移动sqrt(n) n次的话效率为nsqrt(n) ,对于同一块内右端点为有序的那么最多移动 n次  。总的效率为nsqrt(n) 。 要是相邻的查询不同块 最坏效率为O(n) 因为块最多为sqrt(n)个那么坏效率也为nsqrt(n)。   总的效率就为nsqrt(n)

 

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include <iostream>
#define N 200010
#define LL __int64
using namespace std;
struct node {
	int l, r, id;
	int pid;
} s[N];
inline bool cmp(node a, node b) {
	return a.id < b.id || a.id == b.id && a.r < b.r;
}
int a[N];
LL ans[N];
int cnt[1000100];
LL sum = 0;
inline void add(LL x) {
	sum += x * (cnt[x] << 1 | 1);
	cnt[x]++;
}
inline void dec(LL x) {
	sum += x * (1 - (cnt[x] << 1));
	cnt[x]--;
}
int main() {
	int i, j, k, n, m, one;
	memset(cnt, 0, sizeof(cnt));
	scanf("%d%d", &n, &m);
	one = sqrt(n * 1.0);
	for (int i = 1; i <= n; ++i)
		scanf("%d", &a[i]);
	for (int i = 0; i < m; ++i) {
		scanf("%d%d", &s[i].l, &s[i].r);
		s[i].id = s[i].l / one;
		s[i].pid=i;
	}
	sort(s, s + m, cmp);
	int l, r;
	l = s[0].l;
	r = s[0].r;

	for (int i = s[0].l; i <= s[0].r; ++i)
		add(a[i]);
	ans[s[0].pid]=sum;
	for (i = 1; i < m; ++i) {
		while(r<s[i].r)add(a[++r]);
		while(r>s[i].r)dec(a[r--]);
		while(l<s[i].l)dec(a[l++]);
		while(l>s[i].l)add(a[--l]);
		ans[s[i].pid]=sum;
	}
	for(int i=0;i<m;++i)
		printf("%I64d\n",ans[i]);
	return 0;
}

  

CodeForces 86D(Yandex.Algorithm 2011 Round 2),布布扣,bubuko.com

CodeForces 86D(Yandex.Algorithm 2011 Round 2)

原文:http://www.cnblogs.com/L-Ecry/p/3845416.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!