Codeforces Round #533 (Div. 2) C. Ayoub and Lost Array

本文介绍了一种使用动态规划(DP)解决特定数学问题的方法,即计算在给定范围内有多少个长度为n的序列,其元素总和模3等于0。通过定义状态转移方程,利用前缀和的概念,文章详细解释了如何高效地求解此类问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

  题意就是给你一个l和r,要你求一个长度为n的序列有多少种,该序列要满足序列和%3==0.

  定义dp[i][j]为处理到第i个数,前i个数之和模3为j的方案数,设l到r模3为0,1,2的数个数分别为a,b,c

  则有dp[i][0]=(dp[i-1][0]*a%mod+dp[i-1][1]*c%mod+dp[i-1][2]*b%mod)%mod;
        dp[i][1]=(dp[i-1][1]*a%mod+dp[i-1][0]*b%mod+dp[i-1][2]*c%mod)%mod;
        dp[i][2]=(dp[i-1][0]*c%mod+dp[i-1][2]*a%mod+dp[i-1][1]*b%mod)%mod;

 不过这里要注意a,b,c的求解,我就是这里算错wa了好久,以求b为例  l<=3*k+1<=r    -->  1.0*(l-1)/3<=k<=1.0*(r-1)/3   然后又因为k是整数,所有就可以得到k取的是ceil(1.0*(l-1)/3)和floor(1.0*(r-1)/3)之间的整数,然后l到r之间满足l<=3*k+1<=r的k的个数就知道了,那么mod3为1的数的个数也就知道了。

#include<bits/stdc++.h>
using namespace std;
#define fuck(x) cout<<#x<<"  "<<x<<endl;
const int maxn=2e5+10;
long long dp[maxn][3];
const long long mod=1e9+7;
int main()
{
    int n,l,r;
    long long a,b,c;
    scanf("%d %d %d",&n,&l,&r);
    a=r/3-(l-1)/3;
    b=floor(1.0*(r-1)/3)-ceil(1.0*(l-1)/3)+1;
    c=floor(1.0*(r-2)/3)-ceil(1.0*(l-2)/3)+1;
    dp[1][0]=a;
    dp[1][1]=b;
    dp[1][2]=c;
    for(int i=2;i<=n;i++)
    {
        dp[i][0]=(dp[i-1][0]*a%mod+dp[i-1][1]*c%mod+dp[i-1][2]*b%mod)%mod;
        dp[i][1]=(dp[i-1][1]*a%mod+dp[i-1][0]*b%mod+dp[i-1][2]*c%mod)%mod;
        dp[i][2]=(dp[i-1][0]*c%mod+dp[i-1][2]*a%mod+dp[i-1][1]*b%mod)%mod;
    }
    cout<<dp[n][0]<<endl;
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值