P9914 「RiOI-03」匀速相遇 题解

RiOI 的题目都非常赞啊！

做这道题，首先我们思考：对于点 $i$ 和点 $j$，怎么判断它们是否能相遇？（注意：相遇的时刻不一定是整数，坑点）

那么很显然，只要 $\frac{j}{a_i}=\frac{i}{b_j}$ 即可。

根据交叉相乘积相等，上式可转换为 $a_i\times i=b_j\times j$。

那么只需要 $O(n)$ 扫一遍 $a$，统计 $b$ 里面有多少个 $a_i\times i$ 即可。

注意 map 常数比较大，想要卡过去有点难度；所以我们改用 unordered_map。

#include <iostream>
#include <unordered_map>
inline int read(){
	int n=0;
	int f=1;
	char c=getchar();
	while(c<'0' || c>'9'){
		if(c=='-') f=-1;
		c=getchar();
	}
	while(c>='0' && c<='9'){
		n=(n<<3)+(n<<1)+(c^48);
		c=getchar();
	}
	return n*f;
}
inline void write(int x){
	if(x<0){
		putchar('-');
		x=-x;
	}
	if(x>9) write(x/10);
	putchar(x%10^48);
	return;
}
int a[1000005],b[1000005];
std::unordered_map<long long,int> mp;
int main() {
	int n,m; n = read(); m = read();
	for(int i = 1;i <= n;i++) {
		a[i] = read();
	}
	for(int i = 1;i <= m;i++) 
	{
		b[i] = read();
		if(b[i] == 0) continue;
		mp[1ll*b[i]*i]++;
	}
	int cnt = 0;
	for(int i = 1;i <= n;i++) {
		if(a[i] == 0) continue;
		cnt += mp[1ll*a[i]*i];
	}
    write(cnt);
	return 0;
}