网上看到的思路,采用动态规划思想,每一次决策都基于前一次决策的最优解。
即对一个n位数的解都基于前一个n-1位的数的最优解。
我们对一个数的第n位规定一个状态集:即到这一位为止还有几个数字没有使用(我们有 0 1 2 3 共四个数)。
根据规则来说,共有6种状态:
-
0--用了2,剩0,1,3
-
1--用了0,2,剩1,3
-
2--用了2,3,剩0,1
-
3--用了0,1,2,剩3
-
4--用了0,2,3,剩1
- 5--全部用了
于是我们需要让用户输入位数,然后声明同等位数的数组,在每个元素里是6种状态中所包含的该状态下的“符合条件的数”的个数。(是二维数组)
然后用动态规划思想从最小位数开始逐层往上计算。
例:
对于i位状态5的计算,考虑在i-1位时有三种状态可以到达状态5,第3种,此时只能在i位填3,所以*1;第4种,此时只能在i位填1,所以*1;第5种,此时能在i位填2或3(参考规则),所以*2;
states[i][5] = (states[j][3] + states[j][4] + states[j][5] * 2) % mod;
其他同上。
由于采用动态规划,所以取余并没有什么影响。
最后完成计算只需输出i位的第5种状态中的个数。
#include <iostream>
using namespace std;
typedef long long ll;
const int N = 1010;
ll states[N][7];
int main(){
ll mod = 1000000007;
ll n;
cin>>n;
for(ll i =0;i<6;i++)
states[0][i]=0;
/*6种状态
* 0--剩013
* 1--剩13
* 2--剩01
* 3--剩3
* 4--剩1
* 5--无
*/
for(ll i=1;i<=n;i++)
{
ll j = i-1;
states[i][0] = 1;
states[i][1] = (states[j][0] + states[j][1] * 2) % mod;
states[i][2] = (states[j][0] + states[j][2]) % mod;
states[i][3] = (states[j][1] + states[j][3] * 2) % mod;
states[i][4] = (states[j][1] + states[j][2] + states[j][4] * 2) % mod;
states[i][5] = (states[j][3] + states[j][4] + states[j][5] * 2) % mod;
}
cout<<states[n][5]<<endl;
return 0;
}
你的注释的部分的例子中i的位置应该是可以填1或者3。你的确是填写2或者3就错误了
对
妙蛙
这种思路最好理解,但是会不会很慢啊
niu
感觉像是状压dp,合法的状态之间才能转移,这个思路真好 %%%
这不是状态机模型吗
orz
我就知道有数位dp
你好,我想问一下,为什么没有
用了0,剩1,2,3
这一种状态,希望能得到答复。0不可能在首位
如果2在0前面,那就是1号情况了
牛哇,牛哇