牛客网21 F

题目描述
黑妹在玩游戏时又遇到了一个难题，这个游戏初始时给你k个整数x，黑妹需要构建一个长度为n的数组使得该数组恰好包含这k个整数，剩下的位置将由黑妹填上一些不大于100的正整数，要求数组里任意相邻的两个整数互素。
黑妹轻松解决了这个游戏，但是她想知道具体有多少种方法构建这个数组。
由于答案可能很大 , 输出答案对1000000007取模后的值。

思路：枚举右侧数字，需要和左边一位互素。

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std ; 
typedef long long ll ;
int n , x , k ;
const int mod = 1000000007;
ll dp[105][105][55];
int gcd(int x, int y ){
    if(x==0)return y ;
    if(x<y)swap(x,y);
    return gcd(x%y,y);
}
int main(){
    scanf("%d%d%d",&n,&x,&k);
    for(int i = 1 ; i <=100;i++){
        if(i==x)dp[1][i][1]=1;
        else dp[1][i][0]=1;
    }
    for(int i = 2 ;  i<= n ; i++)
        for(int j =1 ; j<= 100 ; j++) 
            for(int v = 0 ; v <=min(i-1 , k );v++)
            for(int s= 1 ; s<= 100 ; s++){
                if(gcd(j,s)>1)continue ; 
                if(j==x)dp[i][j][v+1]=(dp[i][j][v+1]+dp[i-1][s][v])%mod ; 
                else dp[i][j][v]=(dp[i][j][v]+dp[i-1][s][v])%mod ; 
            } 

    ll res=  0 ; 
    for(int i =1 ;i <=100;  i++)
        res=(res+dp[n][i][k])%mod  ;
    printf("%lld\n",res);
    return 0 ; 
}