码迷,mamicode.com
首页 > 其他好文 > 详细

后宫 矩阵快速幂

时间:2017-10-20 18:47:45      阅读:137      评论:0      收藏:0      [点我收藏+]

标签:font   struct   bsp   无限   png   ges   log   name   mod   

[Description]
山山是 2017 级信奥班的成员,因为良好(到大家都嫉妒的程度)的妹子缘而出名。山
山认识的妹子实在是太多,信奥班的各位纷纷猜测山山是怎么做到的。终于,Gemin 揭开了
这个秘密。原来,山山掌握了向妹子脑中写入程序的黑科技。山山向妹子脑中写入的程序是
一个只含有“N”、
“H”两种大写字母的字符串,每个大写字母代表一条指令;指令顺序执
行,并且在执行到结尾后立即返回起始位置,依此无限循环。每条指令都耗费一个指令周期
执行。
两种指令的含义如下:
“N”
:什么都不做;
“H”
:执行到这条指令的妹子 A 会找到一个新的妹子 B,并在妹子 B 的大脑中写入同一段
程序。在下一个指令周期开始时,妹子 B 会加入山山的后宫,并立即开始执行程序。
现在,山山的后宫中只有一个妹子 Eve,在第一个指令周期开始时,Eve 会从头开始执
行程序。现在请计算,在第 N 个指令周期结束时,山山的后宫里有多少妹子。为了避免答
案过大,请输出答案模上 998244353 的值。
[Input]
第一行一个整数:N。
下面一行,一个长度为 L 的、只含有“N”

“H”两种大写字母的字符串 S、代表山山会
在妹子的大脑里写入的程序。
[Output]
一行一个整数,代表在第 N 个指令周期结束时,山山的后宫里有多少妹子。
[Sample]

技术分享

 


说明:什么都不做......

技术分享

 


说明:虽然在第二个周期中,Eve 找到了一个新的妹子,但是新的妹子直到第三个周期的开
始才会加入后宫,因此第二个周期结束时仍然只有一个妹子。

技术分享

 


[Tips]

技术分享


 

我们可以把每个时刻,正在执行第几个命令的妹子数表示出来

for(int i=1;i<=n;i++)

  for(int j=0;j<len;j++)

    if(S[j]==‘N‘)

      f[i+1][j+1]+=f[i][j];

    if(S[j]==‘H‘)

      {

        f[i+1][j+1]+=f[i][j];

        f[i+1][0]+=f[i][j];

      }

我们看到整个递推式,其实很好用矩阵来表示和优化。

首先需要快速幂的矩阵可以确定是一个行数为了len-1,列数为len-1的矩阵,因为每个时刻的每一位数量之和上个时刻的某个数有关。

我们先可以确定矩阵最开始的雏形,由f[i+1][j+1]+=f[i][j]得出,假设了len=4:

0 0 1

1 0 0

0 1 0

而每次碰到H,只有第0行会加上f[i][j],所以我们只用在第0行加上1就可以了。

假如当前字符串是:

NH

那我们可以构造出矩阵:

0 2

1 0

代码:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>

#define ll long long
#define il inline
#define db double

#define mod 998244353

using namespace std;

char S[145],len;

struct mul
{
	ll a[145][145];
	mul ()
	{
		memset(a,0,sizeof(a));
	}
}hehe;

il mul chen(mul x,mul y)
{
	mul p;
	for(int i=0;i<len;i++)
		for(int j=0;j<len;j++)
			for(int k=0;k<len;k++)
				p.a[i][j]=(p.a[i][j]+x.a[i][k]*y.a[k][j])%mod;

	return p;
}

il mul pow(mul x,int b)
{
	mul tmp;
	for(int i=0;i<len;i++)
		for(int j=0;j<len;j++)
			tmp.a[i][j]=0;
	tmp.a[0][0]=1;
	while(b)
		{
			if(b&1)
				tmp=chen(x,tmp);
			x=chen(x,x);
			b>>=1;
		}
	return tmp;
}

int main()
{
	freopen("harem.in","r",stdin);
	freopen("harem.out","w",stdout);

	int n;
	cin>>n;
	
	cin>>S;
	
	len=strlen(S);

	mul T;
	T.a[0][len-1]=1;
	for(int i=1;i<len;i++)
		T.a[i][i-1]=1;

	for(int i=0;i<len;i++)
		if(S[i]==‘H‘)
			T.a[0][i]++;

	mul ans=pow(T,n-1);

	int sum=0;
	for(int i=0;i<len;i++)					 
		sum=(sum+ans.a[i][0])%mod;

	printf("%d\n",sum);
		
	return 0;
}

 

后宫 矩阵快速幂

标签:font   struct   bsp   无限   png   ges   log   name   mod   

原文地址:http://www.cnblogs.com/gshdyjz/p/7700183.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!