785--D

D. Lost Arithmetic Progression

思路

0和-1的情况都挺好分析的。然后A和B的公差必须是lcm==C的公差,最后答案的计数,是先把中间填满了,然后看C的左右两边最大可以拓展多少

代码

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int mod=1e9+7;

//至少有两项,所以d必须是倍数的关系
//大的小的必须要是包含关系

int lcm(int a,int b) {
    return a/__gcd(a,b)*b;
}

signed main() {
    int TT;cin>>TT;
    while(TT--) {
        int a1,b1,c1;cin>>a1>>b1>>c1;
        int a2,b2,c2;cin>>a2>>b2>>c2;
        int d1=a1+(c1-1)*b1,d2=a2+(c2-1)*b2;//尾项
        if(a1>a2||d1<d2||b2%b1||(a2-a1)%b1) {
            //a的首项更大,a的尾项更小,a不是c的倍数,c和a需要重合
            cout<<"0\n";
            continue;
        }
        if(a2-b2<a1||d2+b2>d1) {
            cout<<"-1\n";
            continue;
        }
        int ans=0;
        for(int i=1;i*i<=b2;i++) {
            if(b2%i==0&&lcm(i,b1)==b2)
                ans=(ans+b2/i*b2/i)%mod;//枚举他的公差,反正是需要满足这个条件
            if(b2%i==0&&i*i!=b2) {
                int j=b2/i;
                if(lcm(j,b1)==b2)
                    ans=(ans+b2/j*b2/j)%mod;
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}
posted @ 2022-12-21 21:35  basicecho  阅读(39)  评论(0)    收藏  举报