【CSP】稀疏向量。不用Map,过n=10^9的数据点

题目

思路

在算法设计比赛,我很少用Map。一开始我也没往这边想。如果用直接用数组下标定位,要开10^9那么大的数组,肯定不行。后来一想,直接把两个向量的所有 (index, vlaue) 全部存入一个结构体数组,然后根据index作快排。那么,如果两个向量相同 index 的值不为0,那么经过快排后,在这两个 (index, vlaue) 在结构体数组中一定是相邻元素。因为我们是根据 index 作快排(按index升序或降序都行)。

结构体数组也不需要开10^9那么大,只需要开10^6次方。对 最多10^6 个元素排序,快排的效率还是可以的。

代码

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;

const int maxn = 1e6 + 5;
typedef long long ll;
typedef struct Pos {
	int index, value;
	Pos() {
	}
	Pos(int i, int v) 
	{
		index = i;
		value = v;
	}
	bool operator < (const Pos& p) const
	{
		return this->index < p.index; // 升 
	}
} Pos;

Pos vec[maxn];
//bool cmp(Pos p1, Pos p2)
//{
//	return p1.index < p2.index;
//}
int n, a, b;

int main() 
{
	scanf("%d%d%d", &n, &a, &b);
	int len = a+b;
	for(int i=0; i<len; i++) 
	{
		int t1, t2;
		scanf("%d%d", &t1, &t2);
		vec[i] = Pos(t1, t2);
	}
	
	sort(vec, vec + len);
//	sort(vec, vec + len, cmp);
	
	ll ans = 0;
	for(int i=1; i<len; ) 
	{
		if (vec[i].index == vec[i-1].index) 
		{
			ans += (vec[i].value * vec[i-1].value);
			i += 2;
		}
		else 
			i += 1;
	}
	
	printf("%lld\n", ans);
	
	return 0;	
			

}

 

猜你喜欢

转载自blog.csdn.net/qq_43290318/article/details/108473782