ABC401C

最新推荐文章于 2025-05-03 18:38:58 发布

shy_lihui

最新推荐文章于 2025-05-03 18:38:58 发布

阅读量620

点赞数 33

分类专栏：题解文章标签： c++

本文链接：https://blog.csdn.net/lrz_br/article/details/147660018

版权

题解专栏收录该内容

3 篇文章

订阅专栏

题目大意：

给你两个数 $n$ 和 $k$ ，你要根据 $k$ 求出序列 $a$ ，求 $a_n$ 。

$a_i$ 的计算方式：

考虑递推加前缀和优化， $s_{i}$ 表示 $a$ 前 $i$ 项和，处理 $0\le i < k$ 的时候每次把 $s_i$ 赋值成 $s_{i-1}+1$ 。

处理到 $k\le i$ 时：先把 $s_i$ 赋值成 $s_{i-1}$ ，如果 $i = k$ 就代表 $a_i$ 就是前面所有数的和，直接加上 $s_{i-1}$ 。 $\ne k$ 就是 $s_{i-1}-s_{i-k-1}$ 。

最后 $s_n-s_{n-1}$ 就是 $a_i$ 。

#include<bits/stdc++.h>
#define int long long 
using namespace std;
const int MOD=1e9;
int n,k;
int s[1000005];
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin>>n>>k;
	for(int i=0;i<k;i++)
	{
		s[i]=s[i-1]+1;
		s[i]%=MOD;
	}
	if(k>n)
	{
		cout<<1;
		return 0;
	}
	for(int i=k;i<=n;i++)
	{
		s[i]+=s[i-1]+s[i-1];
		if(i!=k)
		{
			s[i]-=s[i-k-1];
		}
		s[i]%=MOD;
	}
	cout<<(s[n]-s[n-1]+MOD)%MOD;
	return 0;
}