baidu A

题意:
求序列头尾为1的情况,中间n-1个数填充小于等于m的序列的个数

解题思路:
我们考虑DP[N]的情况,
比如{1a1a2....an-11}
那么对于从{a1 到 an-2}我们都有m-1种选择,即
那么对于{an-1}而言,
我们可能有m-1种选择(即它的前一位为1,最后尾也是1),
也可能是m-2种选择(即它的前一位不为1,最后尾是1)
那么我们先让他乘以(m-2),
那么少算的地方在哪里呢?当然是漏掉了前一位为1,最后尾也为1的这个情况
也就是{1a1a2...an-31an-11}
而这时候的{an-1}我们实际上已经遍历了m-2种情况,只剩下一种
自然也就剩下了{1a1a2...an-31}这种情况了,是不是发现了DP公式了?
那么我们可以写出DP公式:
DPn=m-1n-2*(m-2)+DPn-2
等比数列求和就行了
因为N非常大,所以要用nlogn的方法求等比

代码:
const int MO = 1000000007;
long long m, n;
long long powmod(long long a, long long b)
{
long long c = 1;
while (b>0)
{
if (b % 2 != 0)
c = c*a%MO;
a = a*a%MO;
b = b / 2;
}
return c;
}
long long T(long long m, long long n)
{
if (n <= 1) return 1;
long long TN2 = T(m, n / 2);
if (n % 2 == 0)
{
return (TN2 + powmod(m, n / 2) * TN2) % MO;
}
else
{
return (TN2 + powmod(m, n / 2) * TN2 + powmod(m, n - 1)) % MO;
}
}
int main(){
while (cin >> n >> m) {
if (n == 1) {
cout << "1" << endl;
}
else if (n == 2) {
if (m > 1)
cout << "1" << endl;
else
cout << "0" << endl;
}else{
if (m >= 2) {
ll ans = m - 2;
ll a = (m - 1) * (m - 1);
ll nn = (n - 1) / 2;
ans = T(a, nn);
if (n % 2 == 1)
cout << ans * (m - 1) % MO * (m - 2) % MO << endl;
else
cout << ans * (m - 2) % MO << endl;
//cout << sum(a, nn) << endl;
}
else
cout << "0" << endl;
}
}
return 0;
}
#百度#
全部评论
这是我的思路,希望大佬们多多指正。。。
点赞 回复 分享
发布于 2018-09-11 22:12
请问你投的什么岗位c++么?
点赞 回复 分享
发布于 2018-09-11 22:13
s(n) = s(n-1)*(m-2)+s(n-2)*(m-1);  这是普通的通项公式
点赞 回复 分享
发布于 2018-09-11 22:22
由于N非常的大,这个DP你不能直接for一圈,要求出等比通项公式,然后求和,得到答案的方法有两种: 1)直接等比求和公式,然后求逆元做出来 2)二分等比 我是用的二分等比
点赞 回复 分享
发布于 2018-09-11 22:32
楼主你的题意表述有问题吧,表示看不懂
点赞 回复 分享
发布于 2018-09-12 09:59
是这个思路,就是写不出来。。。
点赞 回复 分享
发布于 2018-09-12 10:05

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务