Codeforces 1105C Ayoub and Lost Array

疯狂掉分场的一道dp 几时我才能学会dp啊啊啊啊啊啊啊啊

题意

从l到r中选n个数，这n个数的和是3的倍数的方案数有多少，结果模1e9 + 7。同一个数可以重复取。

思路

dp。先处理出[l, r]区间内模3为0、1、2的数个数，再O(n)跑一遍。转移方程见代码。

代码

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<map>
#include<cmath>
#include<string>
#include<vector>
#include<set>
#include<queue>
#include<stack>
#include<iostream>
#include<algorithm>
using namespace std;

#define debug(x) cout << #x << " = " << x << endl
#define lowbit(x) (x & (-x))

typedef long long LL;
typedef pair<int, int>PII;
const int maxn = 2e5 + 10;
const int inf = 0xfffffff;
const int mod = 1e9 + 7;
const double pi = acos(-1);

LL dp[maxn][5]; //dp[i][j]代表选到第 i 个数时，前 i 个数的和模 3 为 j 的方案数 

inline void open(string s){
	freopen((s + ".in").c_str(), "r", stdin);
	freopen((s + ".out").c_str(), "w", stdout);
}

int main()
{
	LL n, l, r, sum1, sum2, sum0;
	scanf("%lld%lld%lld", &n, &l, &r);
	sum1 = sum2 = sum0 = (r - l + 1) / 3;
	if((r - l + 1) % 3 == 1)
	{
		if(r % 3 == 0) sum0++;
		else if(r % 3 == 1) sum1++;
		else sum2++;
	}
	else if((r - l + 1) % 3 == 2)
	{
		if(r % 3 == 1) sum0++, sum1++;
		else if(r % 3 == 2) sum1++, sum2++;
		else sum0++, sum2++;
	}
	dp[0][0] = 1; //初始化，选 0 个数之和模 3 为 0 的方案数为 1 ，即 1 个都不选
	for(int i = 1; i <= n; i++)
	{
		dp[i][0] = (dp[i - 1][0] * sum0) % mod + (dp[i - 1][1] * sum2) % mod + (dp[i - 1][2] * sum1) % mod;
		dp[i][0] %= mod;
		dp[i][1] = (dp[i - 1][0] * sum1) % mod + (dp[i - 1][2] * sum2) % mod + (dp[i - 1][1] * sum0) % mod;
		dp[i][1] %= mod;
		dp[i][2] = (dp[i - 1][0] * sum2) % mod + (dp[i - 1][1] * sum1) % mod + (dp[i - 1][2] * sum0) % mod;
		dp[i][2] %= mod;
	}
	printf("%lld\n", dp[n][0]);
	return 0;
}