计蒜客 15500 阿里天池的新任务（简单）题解

最新推荐文章于 2019-12-18 22:05:04 发布

DarkoDu

最新推荐文章于 2019-12-18 22:05:04 发布

阅读量728

点赞数

分类专栏：计蒜客文章标签：计蒜客

本文链接：https://blog.csdn.net/m0_37729344/article/details/72599633

版权

计蒜客专栏收录该内容

6 篇文章 0 订阅

订阅专栏

题意

阿里“天池”竞赛平台近日推出了一个新的挑战任务：对于给定的一串 DNA 碱基序列 t，判断它在另一个根据规则生成的 DNA 碱基序列 s 中出现了多少次。

首先，定义一个序列 w：

$\displaystyle w_{i} = \begin{cases}b, & i = 0\\(w_{i-1} + a) \mod n, & i > 0\end{cases}$

接下来，定义长度为 n 的 DNA 碱基序列 s（下标从 0 开始）：

$\displaystyle s_{i} = \begin{cases}A , & (L \le w_{i} \le R) \land (w_{i}\ \mathrm{mod}\ 2 = 0)\\T , & (L \le w_{i} \le R) \land (w_{i}\ \mathrm{mod}\ 2 = 1)\\G , & ((w_{i} < L) \lor (w_{i} > R)) \land (w_{i}\ \mathrm{mod}\ 2 = 0)\\C , & ((w_{i} < L) \lor (w_{i} > R)) \land (w_{i}\ \mathrm{mod}\ 2 = 1)\end{cases}$

其中 $\land$ 表示“且”关系， $\lor$ 表示“或”关系， $a\ \mathrm{mod}\ b$ 表示 a 除以 b 的余数。

现给定另一个 DNA 碱基序列 t，以及生成 s 的参数 n , a , b , L , R，求 t 在 s 中出现了多少次。

思路

根据公式生成s序列，然后进行KMP

代码

#include <cstdio>
#include <cstring>
char s[1000001],t[1000001];
int next[1000001];
int n,a,b,L,R,w,ans,l;
void buildnext()
{
    int i=1,tt=0;
    next[1]=0;
    while(i<l+1)
    {
        while(tt>0&&t[i-1]!=t[tt-1])
            tt=next[tt];
        tt++;
        i++;
        if(t[i-1]==t[tt-1])
            next[i]=next[tt];
        else next[i]=tt;
    }
    while(tt>0&&t[i-1]!=t[tt-1])
        tt=next[tt];
    tt++;
    i++;
    next[i]=tt;
}
int kmp()
{
    int i=0,j=1,n=0;
    int lt,ls;
    lt=strlen(t);
    ls=strlen(s);
    while(lt+1-j<=ls-i)
    {
        if(s[i]==t[j-1])
        {
            i++;
            j++;
            if(j==lt+1)
            {
                n++;
                j=next[j];
            }
        }
        else
        {
            j=next[j];
            if(j==0)
            {
                i++;
                j++;
            }
        }
    }
    return n;
}
int main()
{
    scanf("%d%d%d%d%d",&n,&a,&b,&L,&R);
    scanf("%s",t);
    for(int i=0;i<n;i++)
    {
        if(i==0)
            w=b;
        else w=(w+a)%n;
        if(w>=L&&w<=R&&w%2==0)
            s[i]='A';
        else if(w>=L&&w<=R&&w%2==1)
            s[i]='T';
        else if(w%2==0)
            s[i]='G';
        else s[i]='C';
    }
    //printf("%s\n",s);
    //printf("%s\n",t);
    l=strlen(t);
    buildnext();
    printf("%d\n",kmp());
    return 0;
}