题目
思路
在算法设计比赛,我很少用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;
}