bzoj 4606: [Apio2008]DNA【dp】

写题五分钟读题两小时系列……
看懂题的话不算难，然而我去看了大佬的blog才看懂题……
题目大意是：一个原字符串，其中有一种通配符，合法串的定义是这个串（不含通配符））可以匹配原串并且这个串最多分成k段就能使每一段字典序单调不降。求在所有合法串中字典序第r大的。
设f[i][j][k]表示第i个字符选j，至少需要分成k段的方案数，倒着dp，特判一下通配符，比较基础懒就不多说了
然后对于每个f[i][j]，把k维变成前缀和的形式，因为能分为k段就能分为k+1段。（这里其实当i<k的时候是不对的，但是我写的时候没有特判也过了就没改= =）
然后正着算答案，有点像splay上求k大数的感觉，就是一边匹配一边把r减去当前方案之前（字典序小于当前方案）的方案数。

#include<iostream>
#include<cstdio>
using namespace std;
const int N=50005;
int n,m,a[N],h[305];
long long r,ans,f[N][5][15];
char s[N],b[5];
int main()
{
	h['A']=0,h['C']=1,h['G']=2,h['T']=3,h['N']=4;
	b[0]='A',b[1]='C',b[2]='G',b[3]='T';
	scanf("%d%d%lld%s",&n,&m,&r,s+1);
	for(int i=1;i<=n;i++)
		a[i]=h[s[i]];
	if(a[n]==4)
		f[n][0][1]=1,f[n][1][1]=1,f[n][2][1]=1,f[n][3][1]=1;
	else
		f[n][a[n]][1]=1;
	for(int i=n-1;i>=1;i--)
	{
		if(a[i]==4)
		{
			for(int j=0;j<=3;j++)
				for(int k=1;k<=m;k++)
					for(int l=0;l<=3;l++)
						f[i][j][k]+=f[i+1][l][k-(j>l)];
		}
		else
		{
			for(int k=1;k<=m;k++)
					for(int l=0;l<=3;l++)
						f[i][a[i]][k]+=f[i+1][l][k-(a[i]>l)];
		}
	}
	for(int i=1;i<=n;i++)
		for(int j=0;j<=3;j++)
			for(int k=1;k<=m;k++)
				f[i][j][k]+=f[i][j][k-1];
	for(int i=1,j;i<=n;i++)
	{
		for(j=0;j<=3;j++)
		{
			long long sum;
			if(j<a[i-1])
				sum=f[i][j][m-1];
			else
				sum=f[i][j][m];
			if(r>sum)
				r-=sum;
			else
				break;
		}
		a[i]=j;
		if(a[i]<a[i-1])
			m--;
		printf("%c",b[j]);
	}
	return 0;
}