P8799 [蓝桥杯 2022 国 B] 齿轮

P8799 [蓝桥杯 2022 国 B] 齿轮

分析

最右边的齿轮的转速是最左边齿轮的q倍 == 最右边的齿轮的半径是最左边齿轮的q倍

题意即为:查询数组中是否存在两个数,其中一个是另一个的q倍

题目范围:查询次数q:2*10^5,数组范围2*10^5,所以不能边输入边查询

预处理倍数数组ans[],查询时直接得出答案

注意:

1.遍历数组,看其 j 倍数是否存在

2.倍数为1,单独处理(在输入数组时,若存在b[x]>=2,即至少存在两个相同的数,说明查询倍数为1时能找到答案)

代码

cpp 复制代码
#include<iostream>
#include<algorithm>
using namespace std;

const int N = 200010;
int a[N],b[N],n,q;
bool ans[N],flag;

int main()
{
	scanf("%d %d",&n,&q);
	for(int i = 0;i < n;i ++)
	{
		int x;
		scanf("%d",&x);
		a[i] = x;
		b[x] ++;
		if(b[x] > 1) flag = true;  //单独处理倍数为1时的情况
	}
	if(flag) ans[1] = 1;
	for(int i = 1;i < N;i ++)
	{
		if(b[i])
		{
			for(int j = i * 2;j < N;j += i)
			{
				if(b[j]) ans[j / i] = 1;
			}
		}
	}
	while(q --)
	{
		int x;
		scanf("%d",&x);
		
		if(ans[x]) printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}
相关推荐
CoovallyAIHub19 小时前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破
深度学习·算法·计算机视觉
NAGNIP20 小时前
Serverless 架构下的大模型框架落地实践
算法·架构
moonlifesudo20 小时前
半开区间和开区间的两个二分模版
算法
moonlifesudo20 小时前
300:最长递增子序列
算法
CoovallyAIHub1 天前
港大&字节重磅发布DanceGRPO:突破视觉生成RLHF瓶颈,多项任务性能提升超180%!
深度学习·算法·计算机视觉
CoovallyAIHub1 天前
英伟达ViPE重磅发布!解决3D感知难题,SLAM+深度学习完美融合(附带数据集下载地址)
深度学习·算法·计算机视觉
聚客AI2 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v2 天前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工2 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农2 天前
【React用到的一些算法】游标和栈
算法·react.js