利用矩陣快速冪加速二維dp hdu 5318

題目大意:給出n種串,每種串有無限多個,現在要在這n種串中選擇m個鏈接起來,鏈接的規則是:如果a串的後綴(len >= 2 )是b串的前綴,那麼就可以把b接到a的後面,問最終可以組成多少個不同的串

首先應該排除重複的,因爲重複的不會多產生鏈接。然後找出對於第i種串,後面可以接哪幾個串。

然後dp[i][j],當鏈接了i個串後,以第j個串結尾的有多少種。這樣dp[i][j] = ∑dp[i-1][k] (k串後面可以接j)

其實可以從暴力吧代碼看出三重循環特像矩陣相乘。還有一點需注意:如果狀態轉移是利用if判斷,而不是重新自己構造一個矩陣(只含01表示是否可以轉換),那麼不容易看出是矩陣相乘。因爲矩陣相乘滿足結合律,所有可以有。

因爲m<=1e9,所以用矩陣優化,初始時dp[1][i]全部爲1,計算矩陣temp.a[i][j],如果第i個串後面可以接j,那麼temp.a[i][j] = 1,否則爲0,這樣dp[i]和temp矩陣相乘都會得到dp[i+1],使用矩陣快速冪計算出dp[1]*(temp.a)^(m-1)的和。



copy;http://blog.csdn.net/u011580493/article/details/47111385

思路:

注意字符串去重。還好m不會等於0。= =||

先暴力求出每個串能轉移的位置。a[i][j]爲1,第j個字符串能連在第i個字符串後面;反之爲0,則不能。

定義dp[i][j]:選了i個串,最後以j串結尾的方案數。1

則dp[i][j] += dp[i-1][k]*a[k][j]; (1<=k<= n)。


由於i太大,有1e9個,然後就是矩陣快速冪登場了。

首先res矩陣第一行保存以每個串結尾的方案數,一開始第一行全爲1.,其他位置全爲0。

然後a矩陣快速冪m-1次方(爲什麼要-1,因爲res第一行一開始就設爲1,因此看做已經選了一個串了)。

最後把res的第一行加起來便是要求的答案。



#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<string>
#include<set>
using namespace std;
int vert[55][55];
char a[55][55];
int d[55][55];
long long tmp[55][55];
int ans[55][55];
int n,m;
const int mod = 1e9+7;
void mul(int a[][55],int b[][55])
{
	memset(tmp,0,sizeof(tmp));
	int i,j,k;
	for(i=0;i<n;i++)
	{
		for(j=0;j<n;j++)
		{
			for(k=0;k<n;k++)
			{
				tmp[i][j] += (1LL*a[i][k] * b[k][j])%mod; 
				while(tmp[i][j] >= mod)
				tmp[i][j] -=mod;	
			}
		}
	}
	for(i=0;i<n;i++)
	{
		for(j=0;j<n;j++)
		a[i][j]=tmp[i][j] ;
	}
}

void tcimi(int a[][55],int t)
{
	int i,j;
	for(i=0;i<55;i++)
	{
		ans[i][i]=1;
	}
	while(t)
	{
		if(t%2==1)
		mul(ans,a);
		mul(a,a);
		t/=2;	
	}
	for(i=0;i<55;i++)
	{
		for(j=0;j<55;j++)
		a[i][j] = ans[i][j];
	}
}
void  up(int  &x)
{
	while(x>=mod)
	x-= mod;
}
bool ok(int i,int j)
{
	int dx=2;
	int j1,t;
	int lena=strlen(a[i]);
	while(dx<=strlen(a[j]))
	{
		for(j1=lena-dx,t=0;j1<lena;j1++,t++)
		{
			if(a[i][j1]!=a[j][t])
			break;
		}
		if(j1>=lena)
		{
			return 1;
		}
		dx++;	
	}
	return 0;
}
void vert_init()
{
	int i,j;
	for(i=0;i<n;i++)
	{
		for(j=0;j<n;j++)
		{
			if(ok(i,j))
			vert[i][j]=1;
		}
	}
}
int main()
{
	int i,j,k,t;
	int ans1;
	scanf("%d",&t);
	while(t--)
	{
		set<string > se;
		memset(vert,0,sizeof(vert));
		memset(d,0,sizeof(d));
		memset(ans,0,sizeof(ans));
		scanf("%d%d",&n,&m);
		for(i=0,j=0;i<n;j++,i++)
		{
			//printf("j=%d\n",j);
			scanf("%s",a[j]);
			if(strlen(a[j])<2||se.find(a[j])!=se.end())
			{
				j--;
			}
			se.insert(a[j]);
		}
		n=j;
		/*for(i=2;i<=m;i++)
        {
            for(j=0;j<n;j++)
            {
                for(k=0;k<n;k++)
                {
                    //if(vert[k][j])
                    //{
                        up(d[i][j] += d[i-1][k]*vert[k][j]);    
                    //}    
                }
            }
        }*/
		//printf("n===%d %d\n",n,j);
		vert_init();
		tcimi(vert,m-1);
		for(j=0;j<n;j++)
		{
			d[0][j] = 1; 
		}
		mul(d,vert);
		int ans1=0;
		//for(i=0;i<n;i++)
		for(j=0;j<n;j++)
		up(ans1+=d[0][j]);
		printf("%d\n",ans1);
	}
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章