计数dp

计数dp

　　计数类的$dp$没做过几个,所以之前都放到"思维"标签下了,后来发现原来这属于一类问题啊...搬过来了.

　　

　　管道取珠:https://www.lydsy.com/JudgeOnline/problem.php?id=1566

　　题意概述:

　　

　　有这样的两根管道接在一起,上边有$n$个珠子,下边有$m$个,每次可以从上或下取出一个(向右滑出去),只会有两种颜色的珠子,求不同的取珠序列的总数的平方和.(只要两个取出序列的珠子颜色都相同就视为相同的序列,不用考虑上下).$n,m<=300$

　　$shallwe$:完全可以作为$NOIP$ $D2T1$.

　　既然要求平方和,是不是要先求每种序列的方案数再平方求和呢?然而这样做是做不出来的.

　　出题人怎么说就怎么做是没有前途的...这里有一个挺神奇的转化:视为两个人各拿一个这样的管道取珠,当两个人取到的方案完全相同时答案$+1$,考虑这样做为什么是对的.比如取出某个序列有$a$种方法,那么这第一个人从这$a$种中任取一种,第二个人就有$a$种方法取出与之相同的序列,也就是平方了.太奇妙了.

　　这个转移其实非常简单,考虑一个最普通的:$dp[i][j][k][z]$表示第一个人在上边选了$i$个,下面选了$j$个,第二个人在上边选了$k$个,下边选了$z$个的方案数.既然序列要相同,那两个人取的珠子总数应该是相同的,可以直接删掉任意一维.现在时间复杂度已经能过了,但是空间复杂度又过不了了...显然需要用一个滚动数组,但是这样的状态没有办法滚动,所以重新设计:$dp[s][i][k]$,为了方便理解直接沿用了上边的下标名称,表示共取了$s$个,第一个人在上边取了$i$个,第二个人在上边取了$j$个的方案数.此时可以发现$s$每次只会增加$1$,滚动掉.

　　

 # include <cstdio>
 # include <iostream>
 # include <queue>
 # include <cstring>
 # include <string>
 # define R register int
 # define ll long long
 # define mod 

 using namespace std;

 const int maxn=;
 int n,m;
 char s[maxn];
 int a[maxn],b[maxn];
 int dp[][maxn][maxn];

 int ad (int a,int b)
 {
     a=a+b;
     if(a>=mod) a-=mod;
     return a;
 }

 int main()
 {
     scanf("%d%d",&n,&m);
     scanf("%s",s+);
     for (R i=n;i>=;--i)
         if(s[i]=='A') a[n-i+]=;
     scanf("%s",s+);
     for (R i=m;i>=;--i)
         if(s[i]=='A') b[m-i+]=;
     dp[][][]=;
     for (R s=;s<=n+m;++s)
     {
         int las=s&;
         int no=las^;
         memset(dp[no],,sizeof(dp[no]));
         for (R i=;i<=n;++i)
             for (R k=;k<=n;++k)
                 {
                     int j=s-i,z=s-k,x=dp[las][i][k];
                     if(!x) continue;
                     if(a[i+]==a[k+]) dp[no][i+][k+]=ad(dp[no][i+][k+],x);
                     if(a[i+]==b[z+]) dp[no][i+][k]=ad(dp[no][i+][k],x);
                     if(b[j+]==a[k+]) dp[no][i][k+]=ad(dp[no][i][k+],x);
                     if(b[j+]==b[z+]) dp[no][i][k]=ad(dp[no][i][k],x);}
     }
     printf("%d",dp[(n+m)&][n][n]);
     return ;
 }

管道取珠

　　中国象棋:https://www.lydsy.com/JudgeOnline/problem.php?id=1801

　　题意概述:在一个$n \times m$的棋盘上放任意多个炮,使得它们不能相互攻击,求方案数.$n,m<=100$

　　前置知识:炮的攻击方式

　　一开始以为是炮可以攻击同一行和同一列,然而并不是..."马后炮"了解一下?

　　转化题目:每一行,每一列不能出现超过$2$个棋子的方案数.

　　一开始想到了状压,但是状压压不了$100$位,即使用$bitset$强行压起来了也没法转移.后来发现具体是哪一行有两个棋子或者一个棋子并不重要，我们只需要知道有多少列有两个、一个、没有棋子就可以转移了，因为每列的本质其实是一样的。$dp[i][j][k]$表示前$i$行中$j$列有一个棋子,$k$列有两个棋子的方案数,转移比较麻烦.

　　·如果这一行一个都不放，可以直接继承上面的状态；

　　·如果放一个，可以放到已有一个的地方，那么这一行数量加一，方案数为$(j+1) \times dp[i-1][j+1][k-1]$,也可以放到本来为空的地方$dp[i-1][j-1][k] \times (m-(j-1)-k)$;

　　·如果放两个，可以都放到已有一个的地方$dp[i-1][j+2][k-2] \times c[j+2][2]$，也可以都放到空位上$dp[i-1][j-2][k] \times c[m-(j-2)-k][2]$，也可以一个放到已有一个的地方，另一个放到空位上$dp[i-1][j][k-1] \times (m-j-(k-1)) \times j$。($c$数组中是组合数)

　　

 # include <cstdio>
 # include <iostream>
 # include <cstring>
 # define ll long long
 # define R register int
 # define mod 

 using namespace std;

 int n,m,ans;
 long long dp[][][];
 int c[][];

 int ad (int i,int j,int k)
 {
     long long a=;
     a=dp[i-][j][k]%mod;
     if(j) a=(a+dp[i-][j-][k]*(m-(j-)-k))%mod;
     if(k) a=(a+(dp[i-][j+][k-]*(j+)))%mod;
     if(j>=) a=(a+(dp[i-][j-][k]*c[m-(j-)-k][]))%mod;
     if(k) a=(a+(dp[i-][j][k-]*(m-j-(k-))%mod*j))%mod;
     if(k>=) a=(a+(dp[i-][j+][k-]*c[j+][]))%mod;
     return a;
 }

 int main()
 {
     scanf("%d%d",&n,&m);
     dp[][][]=;
     c[][]=;
     for (R i=;i<=m+;++i)
     {
         c[i][]=;
         for (R j=;j<=;++j)
             c[i][j]=(c[i-][j-]+c[i-][j])%mod;
     }
     for (R i=;i<=n;++i)
         for (R j=;j<=m;++j)
             for (R k=;k+j<=m;++k)
                 dp[i][j][k]=ad(i,j,k);
     for (R i=;i<=m;++i)
         for (R j=;j+i<=m;++j)
             ans=(ans+dp[n][i][j])%mod;
     printf("%d",ans);
     return ;
 }

中国象棋