P8799 [蓝桥杯 2022 国 B] 齿轮

【数论】P8799 [蓝桥杯 2022 国 B] 齿轮

推导：

两个相邻齿轮的线速度是相等的，可得知：$$v_1=v_2=v_3=\cdots =v_n$$又因$v=wr$，故可知$w_1\cdot r_1=w_n\cdot r_n$，变形可得$\frac{w_1}{w_n}=\frac{r_n}{r_1}$

思路1：

由推导可知晓，第一个和最后一个齿轮的转速和半径成反比，也即是找到半径之间的整数商，与所需要的q[i]进行比较，如有相等，便是YES ,否则就是NO。由于整型变量在相除的过程中会取整，无法很好判断，则将每个q[i]与a[i]一次相乘，得到数组temp，用来存储乘积以及和a[i]进行比较。

这么的思路是两层循环用于对比两个数组，时间复杂为O(n^2)，超时啦！

代码：（45’）

// 45‘  超时啦 TLE
#include<iostream>
#include<algorithm>
using namespace std;
int a[200010];
int b[200010];
int m,n;
bool is(int x)
{
    int temp[200010];
    for(int i=0;i<m;i++)
        temp[i]=a[i]*x;
    for(int i=0;i<m;i++)
    {
        if(temp[i]>a[m-1]) continue;
        for(int j=i+1;j<m;j++)
        {
            if(temp[i]==a[j]) return true;
        }
    }
    return false;
}
int main()
{
    cin>>m>>n;

    for(int i=0;i<m;i++) cin>>a[i];
    sort(a,a+m);
    for(int i=0;i<n;i++) cin>>b[i];


    for(int i=0;i<n;i++)
    {
        if(is(b[i])) cout<<"YES";
        else cout<<"NO";
        cout<<'\n';
    }
    return 0;
}

思路2：(利用桶)

逐个列举q[i]的范围过大，不如事先处理好半径数组a[]的数值，利用mp[]数组记录每个半径出现的次数，利用ans[]数组存储半径数组中存在的商，在输入q[i]时，与ans进行对照即可。

遍历一遍数组a[]，第二层循环逐个列举a[i]的倍数，判断是否存在于数组a[]中，若存在，则记录倍数到ans[]中

代码：（AC）

#include<iostream>
#include<algorithm>
#define LL long long
using namespace std;

const LL N=2e5+10;
LL n,Q,k;
LL mp[N],ans[N],a[N];

int main()
{
    cin>>n>>Q;
    for(LL i=1;i<=n;i++)
    {
        cin>>a[i];
        mp[a[i]]++; //记录出现次数
    }
    sort(a+1,a+n+1);

    for(LL i=1;i<n;i++){
        if(i>1&&a[i]==a[i-1]) continue;
        for(LL j=a[i];j<=a[n];j+=a[i]){
            if(!mp[j]) continue;              //如果map里面没有，则跳过
            if(mp[j]==1&&j==a[i]) continue;  //考虑是否存在倍数1
            ans[j/a[i]]=1;
        }
    }

    while(Q--){
        cin>>k;
        if(ans[k]==1) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }

    return 0;
}