舔狗舔到最后一无所有

题目链接:

登录—专业IT笔试面试备考平台_牛客网牛客网是互联网求职神器，C++、Java、前端、产品、运营技能学习/备考/求职题库，在线进行百度阿里腾讯网易等互联网名企笔试面试模拟考试练习,和牛人一起讨论经典试题,全面提升你的技术能力https://ac.nowcoder.com/acm/contest/24213/1002

题目描述

作为队伍的核心，forever97很受另外两个队友的尊敬。
Trote_w每天都要请forever97吃外卖，但很不幸的是宇宙中心forever97所在的学校周围只有3家forever97爱吃的外卖。
如果Trote_w给forever97买了别家的外卖，forever97就会大喊“我不吃我不吃”。
但是forever97又不喜欢连续三天吃一种外卖。
如果Trote_w哪天忘了这件事并且三天给他买了同一家外卖，那么forever97就会把Trote_w的头摁进手机屏幕里。
作为Trote_w的好朋友，你能告诉他连续请forever97吃n天饭，有多少不同的购买方法吗？

输入描述:

多组样例
第一行一个整数T(1<=T<=20)代表测试样例数
接下来t行每行一个整数n，代表Trote_w要请forever97吃n天饭(1<=n<=100000)

输出描述:

输出T个整数代表方案数，由于答案太大，你只需要输出mod 1e9+7 后的答案即可。

示例1

输入

2
3
500

输出

24
544984352

思路一:

题目要求不能连续三天吃同一份外卖,等价于如果前两天吃了同一份的外卖在第三天就不能吃这个餐厅的外卖了,根据动态规划的无后效性,我们前面选择的餐厅是不能影响当前的决策的,因此需要开一个数组去记录前两天吃同一份外卖的情况,于是得到数组arr[n][6],每两个数据记录一种外卖的情况,如arr[n][0]记录正常情况下继承上一天外卖总数,arr[n][1]记录连续吃了两天同种外卖的情况下总数,以三为例

       

	day1	day2	day3
外卖一正常	1	2	6
外卖一不正常	0	1	2
外卖二正常	1	2	6
外卖二不正常	0	1	2
外卖三正常	1	2	6
外卖三不正常	0	1	2

得到动态转移方程:

        for(int j=0;j<6;j++)if(j!=0&&j!=1)arr[i][0]=(arr[i][0]+arr[i-1][j])%mod;
        for(int j=0;j<6;j++)if(j!=2&&j!=3)arr[i][2]=(arr[i][2]+arr[i-1][j])%mod;
        for(int j=0;j<6;j++)if(j!=4&&j!=5)arr[i][4]=(arr[i][4]+arr[i-1][j])%mod;
        arr[i][1]=arr[i-1][0];
        arr[i][3]=arr[i-1][2];
        arr[i][5]=arr[i-1][4];

初始化:

        对于第一天可以选择三种外卖,因此arr[1][0]=arr[1][2]=arr[1][4]=1;

AC代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int mod=1e9+7;
const int maxn=1e5+10;
int t,n;
int arr[maxn][6];
void init(){//数组初始化
    memset(arr,0,sizeof(arr));
    arr[1][0]=arr[1][2]=arr[1][4]=1;
}
int solve(int n){
    int ans=0;
    init();
    for(int i=2;i<=n;i++){
        for(int j=0;j<6;j++)if(j!=0&&j!=1)arr[i][0]=(arr[i][0]+arr[i-1][j])%mod;
        for(int j=0;j<6;j++)if(j!=2&&j!=3)arr[i][2]=(arr[i][2]+arr[i-1][j])%mod;
        for(int j=0;j<6;j++)if(j!=4&&j!=5)arr[i][4]=(arr[i][4]+arr[i-1][j])%mod;
        arr[i][1]=arr[i-1][0];
        arr[i][3]=arr[i-1][2];
        arr[i][5]=arr[i-1][4];
    }
    for(int i=0;i<6;i++)ans=(ans+arr[n][i])%mod;
    return ans;
}
int main(){
    cin>>t;
    while(t--){
        cin>>n;
        cout<<solve(n)<<endl;
    }
    return 0;
}

 思路2

        在思路一的条件下对于每一种外卖其存储的可能情况都相同,因此简化一下,只记录第一个外卖的可能情况.

新的状态转移方程

        arr[i][0]=(2*(arr[i-1][0]+arr[i-1][1]))%mod;
        arr[i][1]=arr[i-1][0];

AC代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define ll long long
const int mod=1e9+7;
const int maxn=1e5+10;
int t,n;
ll arr[maxn][2];
void init(){//数组初始化
    memset(arr,0,sizeof(arr));
    arr[1][0]=1;
}
void test(int n){//测试用的,不删了
    for(int i=0;i<2;i++){
        for(int j=n-1;j<=n;j++){
            cout<<arr[i][j]<<' ';
        }
        cout<<endl;
    }
}
int solve(int n){
    int ans=0;
    init();
    for(int i=2;i<=n;i++){
        arr[i][0]=(2*(arr[i-1][0]+arr[i-1][1]))%mod;
        arr[i][1]=arr[i-1][0];
        //test(i);
    }
    ans=(3*(arr[n][0]+arr[n][1]))%mod;
    return ans;
}
int main(){
    cin>>t;
    while(t--){
        cin>>n;
        cout<<solve(n)<<endl;
    }
    return 0;
}