描述
医学研究者最近发现了某些新病毒,通过对这些病毒的分析,得知它们的DNA序列都是环状的。现在研究者收集了大量的病毒DNA和人的DNA数据,想快速检测出这些人是否感染了相应的病毒。为方便研究,研究者将人的DNA和病毒的DNA均表示成由一些小写字母组成的字符串,然后检测某种病毒的DNA序列是否在患者的DNA序列中出现过,如果出现过,则此人感染了病毒,否则没有感染。注意:人的DNA序列是线性的,而病毒的DNA序列是环状的。
输入
多组数据,每组数据有一行,为序列A和B,A对应病毒的DNA序列,B对应人的DNA序列。A和B都为“0”时输入结束。
输出
对于每组数据输出一行,若患者感染了病毒输出“YES”,否则输出“NO”。
输入样例 1
abbab abbabaab
baa cacdvcabacsd
abc def
0 0
输出样例 1
YES YES NO
思路:
既然是环状的,那么把原来的病毒DNA扩充到之前的二倍,然后从第一个字符开始,依次substr取子串,以此往复原来病毒长度的次数,将每一个字串和主串进行匹配即可。
BFind函数中,i回退到i-(j-0)+1的位置,因为string是从0开始的,不要死记硬背呀!
WA:BFind函数中,如果匹配上了,不能让flag标志=1,而是要进行累加。因为如果简单地用1、0来判断的话,会出现前面所有的字符都不匹配,却只因为最后一个字符匹配而误判YES的情况,要flag++,这样返回到main函数后,如果与子串的长度相等,说明每一个字符都确实匹配上了。
代码:
#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
int BFind(string v, string h)
{
int i = 0, j = 0, flag = 0;
while (i < h.size() && j < v.size())
{
if (v[j] == h[i])
{
i++;
j++;
flag++;//不能写flag=1!!
}
else
{
i = i - j + 1;
j = 0;
flag = 0;
}
}
return flag;
}
int main()
{
string virus, human;
while (cin >> virus >> human && virus != "0" && human != "0")
{
int vlen = virus.length(), ans = 0;
virus += virus;//把这个字符串扩充为原来的两倍,再依次取出长度比对就行了
for (int i = 0; i < vlen; i++)
{
string v = virus.substr(i, vlen);
ans = BFind(v, human);
if (ans == v.length())
{
cout << "YES" << endl;
break;
}
}
if (ans == 0)
cout << "NO" << endl;
}
return 0;
}