Codeforces Round #785 (Div. 2) D. Lost Arithmetic Progression

Codeforces Round #785 (Div. 2) D. Lost Arithmetic Progression

题意： 给出$b,c$两个等差数列数组，输入的$3$个数据分别是开始位置$st$，等差数$d$和数组长度$len$，数组$c$是数组$a\cap b$部分，让我们通过数组$b$和$c$恢复数组$a$，求在满足$b$和$c$前提下$a$数组能构造的数量，如果无解输出 $0$ ，如果无穷多解输出 $-1$ 。

思路： 首先我们得判断什么情况下是无解的什么情况下是无穷多解。

---

无解情况： $dc$不是$db$的倍数，$stc<stb||endc>endb$，$(stb-stc)$%$db!=0$
无穷多解情况： 如果$c$数组首项的前一项$stc-dc<stb$那么数组$a$可以无限向左扩展，同理如果$c$数组尾项的后一项$endc+dc>endb$那么数组$a$可以无限向右扩展。
计算解的个数： 所有的个数和$ans$即为所有$da$为$dc$的约数且$lcm(da,db)=dc$，$a$可以向左右分别扩展$dc/da$个，那么每个满足的$da$带来的贡献即为$dc/da$ * $dc/da$。

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long
int lcm(int a,int b){return a*b/gcd(a,b);}
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int stb,stc,db,dc,lenb,lenc;
const int mod = 1e9+7;
void cf(){//t=100
    cin>>stb>>db>>lenb>>stc>>dc>>lenc;
    int endb,endc;
    endb=stb+(lenb-1)*db;
    endc=stc+(lenc-1)*dc;
    if(endc>endb||stb>stc||dc%db!=0||(stc-stb)%db!=0){//1)无法构造
        cout<<0<<endl;
        return ;
    }
    if(stc-stb<dc||endb-endc<dc){//2)向左向右扩展无限
        cout<<"-1"<<endl;
        return ;
    }
    //sqrt(1e9)
    int ans=0;
    for(int i=1;i*i<=dc;i++){
        if(dc%i==0){
            if(lcm(i,db)==dc){
                ans=(ans+(dc/i)*(dc/i))%mod;
            }
            if(dc/i!=i&&lcm(dc/i,db)==dc){
                ans=(ans+i*i)%mod;
            }
        }
    }
    cout<<ans<<endl;
    return ;
} 
signed main(){
    // freopen("input.txt","r",stdin);
	// freopen("output.txt","w",stdout);
    int t=1;
    cin>>t;
    while(t--){
        cf();
    }
}