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;
}
相关推荐
qystca7 分钟前
洛谷 B3637 最长上升子序列 C语言 记忆化搜索->‘正序‘dp
c语言·开发语言·算法
薯条不要番茄酱7 分钟前
数据结构-8.Java. 七大排序算法(中篇)
java·开发语言·数据结构·后端·算法·排序算法·intellij-idea
今天吃饺子12 分钟前
2024年SCI一区最新改进优化算法——四参数自适应生长优化器,MATLAB代码免费获取...
开发语言·算法·matlab
是阿建吖!13 分钟前
【优选算法】二分查找
c++·算法
王燕龙(大卫)17 分钟前
leetcode 数组中第k个最大元素
算法·leetcode
不去幼儿园1 小时前
【MARL】深入理解多智能体近端策略优化(MAPPO)算法与调参
人工智能·python·算法·机器学习·强化学习
Mr_Xuhhh1 小时前
重生之我在学环境变量
linux·运维·服务器·前端·chrome·算法
盼海2 小时前
排序算法(五)--归并排序
数据结构·算法·排序算法
网易独家音乐人Mike Zhou6 小时前
【卡尔曼滤波】数据预测Prediction观测器的理论推导及应用 C语言、Python实现(Kalman Filter)
c语言·python·单片机·物联网·算法·嵌入式·iot
Swift社区9 小时前
LeetCode - #139 单词拆分
算法·leetcode·职场和发展