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

题目传送门
题意：
一个个数为n的整数列，要你重构，从[l,r]中选取整数，使得最后的序列元素之和是3的倍数。问你整数列有多少种可能，结果mod（1e9 + 7）？
思路：
首先，n<=2e5,1<= l,r<=1e9,注意数据范围。
这是一个多阶段决策问题，所以自然想到DP。两个状态就是徐去了几个数，mod3的余数，所以dp数组是二维的。
dp[i][j]表示选取选取了i个数后的%3余数。
状态转移方程就是选了i个，和的余数为0的状态是由选了i-1个数，余数为0的方法数num[0]（%3 == 0数的个数），余数为1num[2],余数为2*num[1].其他的一次类推，比较好理解。
接下来就是算num[0],num[1],num[2].我这里使用前缀和的思想，不过l-=1.
看代码吧，有些注释会帮助你理解的。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<stack>
#include<queue>
#include<map>
#include<cstring>
#include<string>
#include<cmath>

using namespace std;

typedef long long LL;

#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define pii pair<int,int>
#define all(x) x.begin(),x.end()
#define mem(a,b) memset(a,b,sizeof(a))
#define per(i,a,b) for(int i = a;i <= b;++i)
#define rep(i,a,b) for(int i = a;i >= b;--i)
const int maxn = 2e5;
const int mod = 1e9 + 7;
int n = 0,l = 0,r = 0;
LL dp[maxn+10][4];//要使用longlong，否则WA，结果不对 
int num[4];
void solve(){
	memset(dp,0,sizeof(dp));
	l -= 1;//这里L要减1，便于计算区间之间的个数 
	int le = l % 3,re = r % 3;
	num[0] = r/3 - l/3;
	num[1] = ((re >= 1) ? (r/3+1) : (r/3) ) - ((le < 1) ? l/3 : (l/3 + 1));
	//num[2] = r - l - (num[0] + num[1]);
	num[2] = ((re >= 2) ? (r/3+1) : (r/3)) - ((le < 2) ? l/3 : (l/3+1));
	dp[1][0] = num[0];dp[1][1] = num[1];dp[1][2] = num[2];
	per(i,2,n){
		dp[i][0] = (dp[i-1][0]*num[0]%mod +dp[i-1][1]*num[2]%mod +dp[i-1][2]*num[1]%mod) % mod;
		
		dp[i][1] = (dp[i-1][0]*num[1]%mod +dp[i-1][1]*num[0]%mod + dp[i-1][2]*num[2]%mod) % mod;
		
		dp[i][2] = (dp[i-1][0]*num[2]%mod + dp[i-1][1]*num[1]%mod + dp[i-1][2]*num[0]%mod) % mod;
	}
	printf("%I64d\n",dp[n][0]);
}
int main(){
	while(~scanf("%d %d %d",&n,&l,&r)){
		
		solve();
	}
		
	return 0;
}