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;
}
相关推荐
Yingye Zhu(HPXXZYY)3 小时前
ICPC 2023 Nanjing R L 题 Elevator
算法
程序员Xu6 小时前
【LeetCode热题100道笔记】二叉树的右视图
笔记·算法·leetcode
笑脸惹桃花7 小时前
50系显卡训练深度学习YOLO等算法报错的解决方法
深度学习·算法·yolo·torch·cuda
阿维的博客日记7 小时前
LeetCode 48 - 旋转图像算法详解(全网最优雅的Java算法
算法·leetcode
GEO_YScsn7 小时前
Rust 的生命周期与借用检查:安全性深度保障的基石
网络·算法
程序员Xu8 小时前
【LeetCode热题100道笔记】二叉搜索树中第 K 小的元素
笔记·算法·leetcode
THMAIL9 小时前
机器学习从入门到精通 - 数据预处理实战秘籍:清洗、转换与特征工程入门
人工智能·python·算法·机器学习·数据挖掘·逻辑回归
Kevinhbr9 小时前
CSP-J/S IS COMING
数据结构·c++·算法
蕓晨9 小时前
set的插入和pair的用法
c++·算法
THMAIL10 小时前
深度学习从入门到精通 - AutoML与神经网络搜索(NAS):自动化模型设计未来
人工智能·python·深度学习·神经网络·算法·机器学习·逻辑回归