非010串

如果一个01字符串满足不存在010这样的子串,那么称它为非010串。
求长度为n的非010串的个数。(对1e9+7取模)
Input
一个数n,表示长度。(n<1e15)
Output
长度为n的非010串的个数。(对1e9+7取模)
Input示例
3
Output示例
7

解释:
000
001
011
100
101
110
111
#include <iostream>
#include <cstring>
using namespace std;

const long long int MOD = 1e9+7;

struct matrix
{
	long long int val[4][4];
	void clean()
	{
		memset(val, 0, sizeof(val));
	}

	void init()
	{
		memset(val, 0, sizeof(val));
		for (int i = 0; i < 4; i++)
		{
			val[i][i] = 1;
		}
	}

	matrix operator *(const matrix &a)
	{
		matrix result;
		result.clean();

		for (int i = 0; i < 4; i++)
		{
			for (int j = 0; j < 4; j++)
			{
				for (int k = 0; k < 4; k++)
				{
					result.val[i][j] += val[i][k]*a.val[k][j];
					result.val[i][j] %= MOD;
				}
			}
		}

		return result;
	}
};

matrix fun(long long int n, matrix &base)
{
	matrix result;
	result.init();

	while (n > 0)
	{
		if (n & 1)
		{
			result = result * base;
		}
		base = base * base;
		n >>= 1;
	}

	return result;
}

int main()
{
    long long int n;
    cin >> n;
    
	matrix base = 
			{
			 1, 1, 0, 1,
			 1, 0, 0, 0,
			 0, 1, 0, 0, 
			 0, 0, 1, 0
			};
	matrix result;
	result.clean();
	result.val[0][0] = 7;
	result.val[1][0] = 4;
	result.val[2][0] = 2;
	result.val[3][0] = 1;

	if (n <= 3)
	{
		cout << result.val[3-n][0] << endl;
	}
	else
	{
		base = fun(n-3, base);
		result = base * result;
		cout << result.val[0][0] << endl;
	}
	
    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值