bzoj1494 生成树计数 (dp+矩阵快速幂)

题面欺诈系列...

因为一个点最多只能连到前k个点,所以只有当前的连续k个点的连通情况是对接下来的求解有用的

那么就可以计算k个点的所有连通情况,dfs以下发现k=5的时候有52种。

我们把它们用类似于并查集的方式表达(比如12132代表点1和点3连通,2和5连通,3自己),然后再压缩一下。

但要注意的是,12132和23213这两种实际对应的是一种连通情况,我们只要把它都化成字典序最小的那种就可以了

然后考虑增加一个新点以后状态的转移,可以枚举这个点与前k个点(始状态S)的连边情况,其中有一些是不合法的:

  1.连到了两个本来就连通的点上(导致成环)

  2.在1号点不与其他点连通的情况下,没有连到1号点(导致不连通)

然后再根据连边情况得到终状态E,将trans[S][E]++。最后trans[i][j]表示的就是加入一个点后由i状态到j状态的方案数

那么就可以得到递推式$f[i][j]=\sum^{总状态数}_{k=1}{f[i-1][k]*trans[k][j]}$,其中f[i][j]表示以i为结尾的k个点状态是j的方案数

那么答案就是f[N][1](假设1是都连通的状态)

然后初值就应该是f[K][i]=i状态的方案数,其中i状态的方案数为它其中每个联通块方案数的乘积。

那么联通块的方案数怎么算呢?其实题面已经说了...n个点的方案数就是$n^{n-2)}$

然后就可以愉快地dp啦...

诶?$N<=10^{15}$?

大概能卡过去吧(大雾)

容易发现递推的形式其实和矩阵乘法是相同的,把f[i]和trans看作矩阵,就是$f[N]=f[K]*trans^{N-K}$

然后就可以倍增做矩阵快速幂了。方法和整数的快速幂是一样的。

复杂度:

  K=5的状态数接近50,$log(10^{15})$接近50。

  所以基本上是$50^3*50=6250000$的。

代码写的很蛋疼...

  1 #include<cstdio>
  2 #include<cstring>
  3 #include<algorithm>
  4 #include<map>
  5 #include<vector>
  6 #define LL long long
  7 #define inf 0x3f3f3f3f
  8 using namespace std;
  9 const LL maxn=1e15+5,maxs=55,p65=50000,mod=65521;
 10
 11 LL rd(){
 12     LL x=0;char c=getchar();int neg=1;
 13     while(c<‘0‘||c>‘9‘){if(c==‘-‘) neg=-1;c=getchar();}
 14     while(c>=‘0‘&&c<=‘9‘) x=x*10+c-‘0‘,c=getchar();
 15     return x*neg;
 16 }
 17
 18 LL N;int K,sct,num[6]={0,1,1,3,16,125};
 19 LL f[maxs];
 20 LL trans[2][maxs][maxs],out[2][maxs][maxs];
 21 struct ST{
 22     int s[6];
 23     ST(int x=0){memset(s,x,sizeof(s));}
 24 }sta[maxs];
 25 struct Mat{
 26     LL m[maxs][maxs];
 27     Mat(){memset(m,0,sizeof(m));}
 28 };
 29 int mp[p65];
 30
 31 void print(ST s){
 32     for(int i=1;i<=K;i++) printf("%d",s.s[i]);
 33 }
 34 int STtoInt(ST s){
 35     int x=0;for(int i=1;i<=K;i++) x=x*6+s.s[i];return x;
 36 }
 37
 38 ST toLeast(ST s){
 39     ST flag=ST(-1),re=ST();int n=0;
 40     for(int i=1;i<=K;i++){
 41         if(flag.s[s.s[i]]==-1) flag.s[s.s[i]]=n++;
 42         re.s[i]=flag.s[s.s[i]];
 43     }return re;
 44 }
 45
 46 void dfs1(int x,ST s,int m){
 47     if(x>=K+1){
 48         sta[++sct]=s;mp[STtoInt(s)]=sct;return;
 49     }
 50     for(int i=0;i<=m+1;i++){
 51         s.s[x]=i;
 52         dfs1(x+1,s,max(m,i));
 53     }
 54 }
 55
 56 void dfs2(int x,ST s,ST from){
 57     if(x>=K+1){
 58         bool flag[6],bb=s.s[1];ST to=ST();int mi=5;
 59         memset(flag,0,sizeof(flag));
 60         //print(from);printf("\t");print(s);printf("\n");
 61         for(int i=1;i<=K;i++){
 62             if(!s.s[i]) continue;
 63             if(flag[from.s[i]]) return;
 64             flag[from.s[i]]=1;mi=min(mi,from.s[i]);
 65         }
 66         for(int i=2;i<=K;i++){
 67             to.s[i-1]=flag[from.s[i]]?mi:from.s[i];
 68             if(from.s[1]==to.s[i-1]) bb=1;
 69         }to.s[K]=mi;
 70         if(!bb) return;
 71         to=toLeast(to);
 72         //print(from);printf("\t");print(s);printf("\t");print(to);printf("\n");
 73         trans[0][mp[STtoInt(from)]][mp[STtoInt(to)]]++;
 74
 75     }else{
 76         dfs2(x+1,s,from);s.s[x]=1;dfs2(x+1,s,from);
 77     }
 78 }
 79
 80 void sets(){
 81     for(int i=1;i<=sct;i++){
 82         dfs2(1,ST(),sta[i]);
 83         //for(int j=1;j<=sct;j++) printf("%d ",trans[0][i][j]);printf("\n");
 84         ST cnt=ST();
 85         for(int j=1;j<=K;j++) cnt.s[sta[i].s[j]]++;
 86         f[i]=1;for(int j=0;j<K;j++) if(num[cnt.s[j]])f[i]*=num[cnt.s[j]];
 87     }
 88 }
 89
 90 void solve(){
 91     LL p=N-K;bool b=1,c=1;int i,j,k;
 92     for(i=1;i<=sct;i++) out[0][i][i]=1;
 93     while(p){
 94         if(p&1){
 95             memset(out[c],0,sizeof(out[c]));
 96             for(i=1;i<=sct;i++){
 97                 for(j=1;j<=sct;j++){
 98                     for(k=1;k<=sct;k++){
 99                         out[c][i][j]=(out[c][i][j]+out[c^1][i][k]*trans[b^1][k][j])%mod;
100                     }
101                 }
102             }c^=1;
103         }
104         memset(trans[b],0,sizeof(trans[b]));
105         for(i=1;i<=sct;i++){
106             for(j=1;j<=sct;j++){
107                 for(k=1;k<=sct;k++){
108                     trans[b][i][j]=(trans[b][i][j]+trans[b^1][i][k]*trans[b^1][k][j])%mod;
109                 }
110             }
111         }p>>=1;b^=1;
112     }LL ans=0;
113     for(i=1;i<=sct;i++){
114         ans=(ans+f[i]*out[c^1][i][1])%mod;
115     }printf("%d\n",ans);
116 }
117
118 int main(){
119     int i,j,k;
120     K=rd(),N=rd();
121     dfs1(2,ST(),0);sets();
122     solve();
123     return 0;
124 }

原文地址:https://www.cnblogs.com/Ressed/p/9461115.html

时间: 2024-08-29 06:06:30

bzoj1494 生成树计数 (dp+矩阵快速幂)的相关文章

POJ3735 Training little cats DP,矩阵快速幂,稀疏矩阵优化

题目大意是,n只猫,有k个动作让它们去完成,并且重复m次,动作主要有三类gi,ei,s i j,分别代表第i只猫获得一个花生,第i只猫吃掉它自己所有的花生,第i只和第j只猫交换彼此的花生.k,n不超过100,m不超过1000,000,000,计算出最后每只猫还剩下多少个花生. 我们假设一个n维向量P,每个分量的值代表这n只猫所拥有的花生数,那么对于gi操作其实就是在第i维分量上加上1:对于ei,那就是在第i维分量上乘以0,说到这里,有木有感觉这很像3D坐标转化中的平移矩阵和缩放矩阵?没错,就是这

HDU 2294 Pendant (DP+矩阵快速幂降维)

HDU 2294 Pendant (DP+矩阵快速幂降维) ACM 题目地址:HDU 2294 Pendant 题意: 土豪给妹子做首饰,他有K种珍珠,每种N个,为了炫富,他每种珍珠都要用上.问他能做几种长度[1,N]的首饰. 分析: 1 ≤ N ≤ 1,000,000,000简直可怕. 首先想dp,很明显可以想到: dp[i][j] = (k-(j-1))*dp[i-1][j-1] + j*dp[i-1][j](dp[i][j]表示长度为i的并且有j种珍珠的垂饰有多少个) 然后遇到N太大的话,

POJ3420 Quad Tiling DP + 矩阵快速幂

题目大意是用1*2的骨牌堆积成4*N的矩形,一共有多少种方法,N不超过10^9. 这题和曾经在庞果网上做过的一道木块砌墙几乎一样.因为骨牌我们可以横着放,竖着放,我们假设以4为列,N为行这样去看,并且在骨牌覆盖的位置上置1,所以一共最多有16种状态.我们在第M行放骨牌的时候,第M+1行的状态也是有可能被改变的,设S(i,j)表示某一行状态为i时,将其铺满后下一行状态为j的方案书.考虑下如果我们让矩阵S和S相乘会有什么意义,考虑一下会发现S*S的意义当某行状态为i,接着其后面第2行的状态为j的可行

HDU5863 cjj&#39;s string game(DP + 矩阵快速幂)

题目 Source http://acm.split.hdu.edu.cn/showproblem.php?pid=5863 Description cjj has k kinds of characters the number of which are infinite. He wants to build two strings with the characters. The lengths of the strings are both equal to n. cjj also def

HDU 5434 Peace small elephant 状压dp+矩阵快速幂

题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=5434 Peace small elephant Accepts: 38 Submissions: 108 Time Limit: 10000/5000 MS (Java/Others) Memory Limit: 65536/65536 K (Java/Others) 问题描述 小明很喜欢国际象棋,尤其喜欢国际象棋里面的大象(只要无阻挡能够斜着走任意格),但是他觉得国际象棋里的大象太凶残了,于是他

hdu 4878 ZCC loves words(AC自动机+dp+矩阵快速幂+中国剩余定理)

hdu 4878 ZCC loves words(AC自动机+dp+矩阵快速幂+中国剩余定理) 题意:给出若干个模式串,总长度不超过40,对于某一个字符串,它有一个价值,对于这个价值的计算方法是这样的,设初始价值为V=1,假如这个串能匹配第k个模式串,则V=V*prime[k]*(i+len[k]),其中prime[k]表示第k个素数,i表示匹配的结束位置,len[k]表示第k个模式串的长度(注意,一个字符串可以多次匹配同意个模式串).问字符集为'A'-'Z'的字符,组成的所有的长为L的字符串,

UVA 11651 - Krypton Number System(DP+矩阵快速幂)

UVA 11651 - Krypton Number System 题目链接 题意:给一个进制base,一个分数score求该进制下,有多少数满足一下条件: 1.没有连续数字 2.没有前导零 3.分数为score,分数的计算方式为相邻数字的平方差的和 思路:先从dp入手,dp[i][j]表示组成i,最后一个数字为j的种数,然后进行状态转移,推出前面一步能构成的状态,也就是到dp[(b - 1) * (b - 1)][x]. 然后可以发现后面的状态,都可以由前面这些状态统一转移出来,这样就可以利用

Codeforces Round #291 (Div. 2) E - Darth Vader and Tree (DP+矩阵快速幂)

这题想了好长时间,果断没思路..于是搜了一下题解.一看题解上的"快速幂"这俩字,不对..这仨字..犹如醍醐灌顶啊...因为x的范围是10^9,所以当时想的时候果断把dp递推这一方法抛弃了.我怎么就没想到矩阵快速幂呢.......还是太弱了..sad..100*100*100*log(10^9)的复杂度刚刚好. 于是,想到了矩阵快速幂后,一切就变得简单了.就可以把距离<=x的所有距离的点数都通过DP推出来,然后一个快速幂就解决了. 首先DP递推式很容易想到.递推代码如下: for(

瓷砖铺放 (状压DP+矩阵快速幂)

未加矩阵快速幂 50分 1 const dx:array[1..8,1..3] of longint= 2 ((-1,0,0),(-1,0,0),(1,0,0),(0,1,0),(-1,0,0),(-1,1,0),(0,1,0),(-1,0,1)); 3 dy:array[1..8,1..3] of longint= 4 ((0,1,0),(0,-1,0),(0,-1,0),(1,0,0),(0,1,-1),(0,0,-1),(1,0,-1),(0,1,0)); 5 mo=65521; 6 va