codeforces(D. Nezzar and Board) 裴蜀定理

原題鏈接
在這裏插入圖片描述
n個數,每次可以從數組中選擇一個x和一個y,將2*x-y添加到原數組中,問是否能構造出k

仔細想一下,最後的k,一定是某一個基數x[i] 加上 若干任一一對數的差的和
因此對於每個k-x[i],若可以整除gcd(x[2]-x[1],x[3]-x[2],…x[n]-x[n-1]) 即爲YES
代碼:

//#pragma GCC optimize("Ofast")
//#pragma GCC target("avx,avx2,fma")
//#pragma GCC optimization ("unroll-loops")
#include <bits/stdc++.h>
#define pb push_back
#define ll long long
#define IOS std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;

ll t,n,k;
ll a[200005];
int main()
{
   
   
    cin>>t;
    while(t--)
    {
   
   
        cin>>n>>k;
        for(int i=1;i<=n;i++)
        {
   
   cin>>a[i];}
        ll op=a[2]-a[1];
        for(int i=3;i<=n;i++)
        {
   
   
            op=__gcd(a[i]-a[i-1],op);
        }
        bool flag=0;
        for(int i=1;i<=n;i++)
        {
   
   
            ll cc=k-a[i];
            if(cc%op==0){
   
   flag=1;break;}
        }
        if(flag==1){
   
   cout<<"YES"<<endl;}
        else{
   
   cout<<"NO"<<endl;}
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章