[hihocoder1318]非法二进制数

问题描述

时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述
如果一个二进制数包含连续的两个1，我们就称这个二进制数是非法的。

小Hi想知道在所有 n 位二进制数（一共有2n个）中，非法二进制数有多少个。

例如对于 n = 3,有 011, 110, 111 三个非法二进制数。

由于结果可能很大，你只需要输出模109+7的余数。

输入
一个整数 n (1 ≤ n ≤ 100)。

输出
n 位非法二进制数的数目模109+7的余数。

样例输入
3
样例输出
3

算法详解

首先求解合法的二进制数，然后用整体数目减去合法数目就可以得到非法二进制数。

合法的二进制数目

使用动态规划算法，i位二进制数中分别以0和1的合法二进制数目，分别用cnt[i][0]以及cnt[i][1]表示。转移方程是cnt[i][0] = cnt[i-1][0]+cnt[i-1][1]；cnt[i][1] = cnt[i-1][0]。

模数的细节处理

最后需要总体二进制数目减去合法二进制数，因为是在模数域下的，所以可以存在减数大于被减数的情况，需要显示加上MOD处理。同时因为所有数目要能够存储2×MOD的数，所以需要unsigned long long int。

全部代码

#include <iostream>

using namespace std;

#define MOD 1000000007

unsigned long long int cnt[102][2];

void print_cnt(unsigned long long int n){
    cout << "CNT: " << endl;
    for (unsigned long long int i = 0;i < n;++i)
        cout << cnt[i][0] << ' ' << cnt[i][1] << endl;
    cout << endl;
}

int main()
{
    unsigned long long int n;
    cin >> n;

    cnt[0][0] = cnt[0][1] = 1;
    for (unsigned long long int i = 1;i < n;++i) {
        cnt[i][0] = (cnt[i-1][0] +cnt[i-1][1]) % MOD;
        cnt[i][1] = cnt[i-1][0];
    }


    unsigned long long int total = 1;
    for (unsigned long long int i = 0;i < n;++i)
        total = (total*2) % MOD;

    unsigned long long int total_legal = (cnt[n-1][0]+cnt[n-1][1])%MOD;
    if (total < total_legal)
        total += MOD;

    cout << total - total_legal << endl;
    return 0;
}