[BZOJ4606][Apio2008]DNA-动态规划

[Apio2008]DNA

Description

分析如DNA序列这样的生命科学数据是计算机的一个有趣应用。从生物学的角度上说,DNA 是一种由腺嘌呤、胞嘧啶、鸟嘌呤和胸腺嘧啶这四种核苷酸组成的链式结构。这四种核苷酸分别用大写字母A、C、G、T表示。这样,一条DNA单链可以被表示为一个只含以上四种字符的字符串。我们将这样的字符串称作一个DNA序列 。

有时生物学家可能无法确定一条DNA单链中的某些核苷酸。在这种情况下,字符N将被用来表示一个不确定的核苷酸。换句话说,N可以用来表示A、C、G、T中的任何一个字符。我们称包含一个或者多个N的DNA序列为未完成序列;反之,就称作完成序列。如果一个完成序列可以通过将一个未完成序列中的每个N任意替换成A、C、G、T得到的话,就称完成序列适合这个未完成序列。举例来说,ACCCT适合ACNNT,但是AGGAT不适合。

研究者们经常按照如下方式排序四种核苷酸:A优先于C,C优先于G,G优先于T。如果一个DNA序列中的每个核苷酸都与其右边的相同或者优先,就将其归类为范式-1。举例来说,AACCGT是范式-1,但是AACGTC不是。

一般来说,一个DNA序列属于范式-j(j>1),只要它属于范式-(j-1)或者是一个范式-(j-1)和一个范式-1的连接。举例来说,AACCC、ACACC和ACACA都是范式-3,但GCACAC和ACACACA不是。

同样,研究者们按照字典序对 DNA 序列进行排序。按照这个定义,最小的属于范式-3的DNA序列是AAAAA,最大的是TTTTT。这里是另外一个例子,考虑未完成序列 ACANNCNNG。那么前7个适合这个未完成序列的DNA序列是:
ACAAACAAG
ACAAACACG
ACAAACAGG
ACAAACCAG
ACAAACCCG
ACAAACCGG
ACAAACCTG

写一个程序,找到按字典序的第R个适合给定的长度为M的未完成序列的范式-K。

Input

输入第一行包含三个由空格隔开的整数:M(1≤M≤50,000),K(1≤K≤10)和R(1≤R≤2×10^12)。
第二行包含一个长度为M的字符串,表示未完成序列。
保证适合该未完成序列的范式-K的总数不超过4×10^18
因此该数可以用C和C++中的long long类型或者Pascal中的Int64类型表示。
同时,R不会超过适合给定未完成序列的范式-K的总数。

Output

在第一行中输出第R个适合输入中的未完成序列的范式-K。

Sample Input

9 3 5
ACANNCNNG

Sample Output

ACAAACCCG

---

细节无数高能预警注意!!!
这是一个不难想，但较难写的DP……
咱在过了样例之后就直接A掉了呢……但是……
在样例都过不了的状态下调了好久……

(╯‵□′)╯︵┻━┻

---

思路:
这题看着就像钦定DP……
想象一下一棵字典树的样子，N处分叉。
然后，这题的DP就像是在这棵字典树上跑带转折这个条件的树形DP……
大概就是这个意思吧……

定义转折:|范式|-1。(因为基础是-1)
于是令f[i][j][k]表示当前要放第i个，放的是第j种核苷酸，有k个转折的方案数。
因为根据上面的比方，是在字典树上跑。(不是说真的要用字典树，这是比方)
所以，从后往前，得到递推式:
若当前位置字符与j相同或为’N’:
f[i][j][k]=sum(l=1~j-1){f[i+1][l][k+1]}+sum(l=j~4){f[i+1][l][k]}

咱还需要一个sum数组:
sum[i][j][k]=sum(l=1~k){f[i][j][l]}

最后在sum上当做字典树跑即可~

即为:
从前往后对每个位置从’A’到’T’进行询问，若r>当前sum，r-=当前sum~
否则往下走一层~
沿路记录下从何处往下，输出，完成ヾ(o◕∀◕)ﾉヾ~

#include<iostream>#include<cstdio>#include<cstring>#include<cstdlib>#include<algorithm>using  namespace std;typedef long long ll;inline int trans(char c){    switch(c)    {        case 'N':return 0;break;        case 'A':return 1;break;        case 'C':return 2;break;        case 'G':return 3;break;        case 'T':return 4;break;    }}inline char decode(int a){    switch(a)    {        case 1:return 'A';break;        case 2:return 'C';break;        case 3:return 'G';break;        case 4:return 'T';break;    }}const int M=50233;int m,k;int base[M],ans[M];ll r,f[M][5][13],sum[M][5][13];int main(){    char s[M];    scanf("%d%d%lld",&m,&k,&r);    k--;    scanf("%s",s);    for(int i=1;i<=m;i++)        base[i]=trans(s[i-1]);    f[m+1][4][0]=1;    for(int i=m;i>=1;i--)//pos        for(int j=1;j<=4;j++)//acgt        {            if(!base[i] || base[i]==j)            {                for(int jj=1;jj<j;jj++)//last acgt                    for(int l=1;l<=k;l++)//turns                        f[i][j][l]+=f[i+1][jj][l-1];                for(int jj=j;jj<=4;jj++)                    for(int l=0;l<=k;l++)                        f[i][j][l]+=f[i+1][jj][l];            }            else                for(int l=0;l<=k;l++)                    f[i][j][l]=0;            sum[i][j][0]=f[i][j][0];            for(int l=1;l<=k;l++)                sum[i][j][l]=f[i][j][l]+sum[i][j][l-1];        }    ans[0]=1;    for(int i=1,j;i<=m;i++)    {        k--;        for(j=1;j<=4;j++)        {            if(j==ans[i-1])                k++;            if(sum[i][j][k]>=r)                break;            r-=sum[i][j][k];        }        ans[i]=j;        putchar(decode(j));    }    putchar('\n');    return 0;}