牛牛与数组 (简单dp)

题目链接
这种题一看就是dp啊,dp[i][j]表示第i位放j的方案数,转移方程为dp[i][j]=dp[i-1][k]{k<=i||k%i!=0},当然我们可以三层循环来找,但数据显然会超时,那么我们只能在第二层循环中用中间变量记录一下可以省去一层循环,但是为倍数的情况必须要考虑,所以先预处理出所有的倍数,然后dp的过程中减去倍数的情况即可,总体复杂度O(n* k * sqrt(k)。

#include <set>
#include <map>
#include <queue>
#include <stack>
#include <math.h>
#include <bitset>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#define MAXN 1010100
#define LL long long
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll __int64
#define INF 0x7fffffff
#define cs(s) freopen(s,"r",stdin)
#define mem(x) memset(x,0,sizeof(x))
#define PI acos(-1)
#define eps 1e-10
using namespace std;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int lcm(int a,int b){return a/gcd(a,b)*b;}
LL powmod(LL a,LL b,LL MOD){LL ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}
//head
const LL mod=1e9+7;
LL dp[11][100003];
vector<LL>p[100001];
LL n,k;
void P(){//预处理倍数
	for(int i=1;i<=k;i++){
		for(int j=2;1ll*j*i<=k;j++){
			p[i].pb(j*i);
		}
	}
}
LL a[111];
int main(){
	ios::sync_with_stdio(false);
	cin>>n>>k;
	P();
	for(int i=1;i<=k;i++)dp[1][i]=1;
	for(int i=1;i<=n;i++)dp[i][1]=1;
	for(int i=2;i<=n;i++){
		LL pr=0;
		for(int j=1;j<=k;j++)pr=(0ll+pr+dp[i-1][j])%mod;
		LL sum=dp[i-1][1];
		for(int j=2;j<=k;j++){
			LL mi=pr;
			dp[i][j]=(0ll+sum+dp[i-1][j])%mod;
			sum=(0ll+sum+dp[i-1][j])%mod;//为dp[i-1][1]到dp[i-1][j]的和
			mi-=sum;
			for(int x:p[j]){
				mi-=dp[i-1][x];//减去倍数
			}
			dp[i][j]+=mi;
		}
	}
	LL ans=0;
	for(int i=1;i<=k;i++)ans=(ans+dp[n][i])%mod;
		cout<<ans;
	return 0;
}
全部评论

相关推荐

菜菜咪:1. 可以使用简历网站的模版,美观度会更好一点 2. 邮箱可以重新申请一个,或者用qq邮箱的别名,部分hr可能会不喜欢数字邮箱 3. 项目经历最好分点描述,类似的项目很多,可以参考一下别人怎么写的 4. 自我评价可加可不加,技术岗更看重技术。最后,加油,优秀士兵
点赞 评论 收藏
分享
尊嘟假嘟点击就送:加v细说,问题很大
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务