「ZJOI2012」波浪 DP

「ZJOI2012」波浪

题解部分

由于本题中要求绝对值，但我们可以选定一个顺序加点以去掉绝对值（这里就用从小到大加点）。就可以算出每个点对答案的贡献。（这里的答案不是指方案数，而是指相邻数差的绝对值的和）

那么对于一个状态，我们只用知道它有几个联通块，即可，那么它插的位置就是联通块位置+1，然后再分类讨论它是否与两边的联通块相连即可，还有放在边界上的情况。

然后我们每加入一个点，它插入的位置有两种可能，插在边界或不插在边界（也就是它最后在不在边上）。

若它插在边界上，又分有两种可能。

如果插入时它的旁边有一个点（此时之前肯定已经加了一个点了）（那么这个点的权值比它小），那么它对答案的贡献为$i$，此时它可以插的位置是2-原有边界数量。否则它的旁边没有点（那么最后它旁边的点权值比它大），它对答案的贡献就是$-i$，这是它可以插的位置也是2-原有边界的数量。

要注意边界也算一个联通块。

若它没插在边界上，就又分有三种可能。

如果插入的时候它的旁边没有点（可以插入的位置为联通块个数$+1$-边界个数（只能在边界的一边放数），那么它对答案的贡献是$-2\ast i$，如果插入的时候它的旁边有一个点（这个就是插在一个联通块的旁边，故可以插入的位置就是联通块个数$\ast 2$-边界个数），那么它对答案的贡献就是$0$。如果插入的时候它的旁边都有点（那么两个联通块就会合成一个，可以插入的位置就是联通块个数$-1$），它对答案的贡献就是$2\ast i$

故我们对于一个状态只用知道它有几个联通块，当前答案，已经决策的边界数量。即可进行转移。

K<=8的时候用double（快），K>8的时候用float128（不想写高精度）。

#include<cstdio>
#include<bits/stdc++.h>
#include<cmath>
#include<algorithm>
#include<cstring>
#define M 35
#define N 5000
using namespace std;
int n,m,K;
int num[M];
template <class T>
void Print(T ans){//输出要考虑进位，答案可能是1.00000 
	num[0]=0;
	ans=ans*10;
	for(int i=1;i<K;i++){
		num[i]=(int)ans;
		ans=(ans-num[i])*10;
	}
	num[K]=(int)(ans+0.5);
	for(int i=K;i>=1;i--)if(num[i]>=10)num[i]-=10,num[i-1]++;
	printf("%d.",num[0]);
	for(int i=1;i<=K;i++)printf("%d",num[i]);
	puts("");
}
template <class T>//这个不会用(实际我也看不懂，但为了偷懒QWQ）可以直接上两份代码 
void Solve(T dp[][105][10005][3]){
	dp[1][0][N][0]=1;
	int cur=0;
	for(int i=1;i<=n;i++,cur=!cur){
		int lim=i*(i-1);
		for(int j=0;j<i;j++)for(int k=max(N-lim,0);k<=min(N+lim,N<<1);k++)for(int l=0;l<=2;l++)dp[cur][j][k][l]=0;
		//这样清空会快很多 
		for(int j=0;j<i;j++){
			for(int k=max(N-lim,0);k<=min(N+lim,N<<1);k++){
				for(int l=0;l<=2;l++){
					if(!dp[!cur][j][k][l])continue;
					if(j+1-l>0)dp[cur][j+1][k-2*i][l]+=dp[!cur][j][k][l]*(j+1-l);//两边没数 
					if(j*2>l)dp[cur][j][k][l]+=dp[!cur][j][k][l]*(j*2-l);//一边有数 
					if(j>=1)dp[cur][j-1][k+2*i][l]+=dp[!cur][j][k][l]*(j-1);//两边有数 
					if(l<=2)dp[cur][j][k+i][l+1]+=dp[!cur][j][k][l]*(2-l);//放在边界，边界旁有数 
					if(l<=2)dp[cur][j+1][k-i][l+1]+=dp[!cur][j][k][l]*(2-l);//放在边界，边界旁没数 
				}
			}
		}
	}
	T ans=0;
	for(int i=m+N;i<=N+N;i++)ans=ans+dp[!cur][1][i][2];
	for(int i=1;i<=n;i++)ans=ans/i;
	Print(ans);
}
double dp1[2][105][10005][3];//已经枚举到第i位，有j个联通块，答案为k，已经枚举了l个边界 
__float128 dp2[2][105][10005][3];
int main(){
	scanf("%d%d%d",&n,&m,&K);
	if(K<=8)Solve(dp1);
	else Solve(dp2);
	return 0;
}