RiOI 的题目都非常赞啊!
做这道题,首先我们思考:对于点 i i i 和点 j j j,怎么判断它们是否能相遇?(注意:相遇的时刻不一定是整数,坑点)
那么很显然,只要 j a i = i b j \large\frac{j}{a_i}=\frac{i}{b_j} aij=bji 即可。
根据交叉相乘积相等,上式可转换为 a i × i = b j × j a_i\times i=b_j\times j ai×i=bj×j。
那么只需要 O ( n ) O(n) O(n) 扫一遍 a a a,统计 b b b 里面有多少个 a i × i a_i \times i ai×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;
}