牛客 21738 牛牛与数组

题目链接

题目描述

牛牛喜欢这样的数组：
1：长度为n
2：每一个数都在1到k之间
3：对于任意连续的两个数A，B，A<=B 与（A % B != 0） 两个条件至少成立一个

请问一共有多少满足条件的数组，对1e9+7取模
输入描述:
输入两个整数n,k

1 ≤ n ≤ 10
1 ≤ k ≤ 100000
输出描述:
输出一个整数

输入
2 2
9 1
3 3
2 1234

输出
3
1
15
1515011

具体思路
一开始大致知道状态转移的思路是什么，但不知为何写着写着把自己绕进去了，哭死了。设立一个二维数组num[i][j]表示在有i位以j为结尾符合要求的数组个数，其状态转移由num[i-1][]的所有1-k数组在末尾加上j的方案数求得，即求和num[i-1][1]到num[i-1][k]，因为其需要满足题目的要求，即必须还要减去倍数情况下的个数，这样的状态转移方程就出来了，最后求得的num数组是在有i位以j为结尾符合要求的数组个数，所以还需要将第n位的所有1-k的num数组求和起来，结果取余mood

#include<stdio.h>
#include<math.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int max_k=1e5+50,mood=1e9+7;
int num[15][max_k];
int main()
{
    int n,k;
    scanf("%d %d",&n,&k);
    for (int i=1;i<=k;i++)
    {
        num[1][i]=1;
    }
    int sum,notsum;
    for (int i=2;i<=n;i++)
    {
        sum=0;
        for (int j=1;j<=k;j++)
        {
            sum=(sum+num[i-1][j])%mood;
        }
        for (int j=1;j<=k;j++)
        {
            notsum=0;
            for (int l=2*j;l<=k;l=l+j)
            {
                notsum=(notsum+num[i-1][l])%mood;
            }
            num[i][j]=(sum-notsum+mood)%mood;
        }
    }
    int ans=0;
    for (int i=1;i<=n;i++)
    {
        for (int j=1;j<=k;j++)
        {
            printf("%d ",num[i][j]);
        }
        printf("\n");
    }
    for (int i=1;i<=k;i++)
    {
        ans=(ans+num[n][i])%mood;
    }
    printf("%d",ans);
    return 0;
}