牛客寒假算法基础集训营2 G 处女座与复读机（模拟）

链接：https://ac.nowcoder.com/acm/contest/327/G
来源：牛客网
 

题目描述

一天，处女座在牛客算法群里发了一句“我好强啊”，引起无数的复读，可是处女座发现复读之后变成了“处女座好强啊”。处女座经过调查发现群里的复读机都是失真的复读机，会固定的产生两个错误。一个错误可以是下面的形式之一：

1.       将任意一个小写字母替换成另外一个小写字母

2.       在任意位置添加一个小写字母

3.       删除任意一个字母

处女座现在在群里发了一句话，他收到了一个回应，他想知道这是不是一个复读机。

输入描述:

两行
第一行是处女座说的话s
第二行是收到的回应t
s和t只由小写字母构成且长度小于100

输出描述:

如果这可能是一个复读机输出”YES”，否则输出”NO”

示例1

输入

复制

abc
abcde

输出

复制

YES

说明

abc->abcd->abcde

示例2

输入

复制

abcde
abcde

输出

复制

YES

说明

abcde->abcdd->abcde

备注:

只要能经过两步变换就从s得到t就有可能是复读机。

#include <bits/stdc++.h>
using namespace std;
string a,b;
 
int main()
{
    cin>>a>>b;
    if (fabs((int)a.length()-(int)b.length())>2)
    {
        puts("NO");
    }
    else
    {
        if (b.length()==a.length()+2)
        {
            int can=0;
            for (int i=0;i<b.length();i++)
            {
                for (int j=i+1;j<b.length();j++)
                {
                    string s="";
                    for (int k=0;k<b.length();k++)
                    {
                        if (k!=i && k!=j) s.push_back(b[k]);
                    }
                    if (a==s) can=1;
                }
            }
            if (can==1) puts("YES");
            else puts("NO");
        }
        else if ((int)b.length()==(int)a.length()-2)
        {
            swap(a,b);
            int can=0;
            for (int i=0;i<b.length();i++)
            {
                for (int j=i+1;j<b.length();j++)
                {
                    string s="";
                    for (int k=0;k<b.length();k++)
                    {
                        if (k!=i && k!=j) s.push_back(b[k]);
                    }
                    if (a==s) can=1;
                }
            }
            if (can==1) puts("YES");
            else puts("NO");
        }
        else if (b.length()==a.length()+1)
        {
            int can=0;
            for (int i=0;i<b.length();i++)
            {
                string s="";
                for (int j=0;j<b.length();j++)
                {
                    if (j!=i) s.push_back(b[j]);
                }
                int cnt=0;
                for (int i=0;i<a.length();i++)
                {
                    if (a[i]!=s[i]) cnt++;
                }
                if (cnt<=1) can=1;
            }
            if (can==1) puts("YES");
            else puts("NO");
        }
        else if ((int)b.length()==(int)a.length()-1)
        {
            swap(a,b);
            int can=0;
            for (int i=0;i<b.length();i++)
            {
                string s="";
                for (int j=0;j<b.length();j++)
                {
                    if (j!=i) s.push_back(b[j]);
                }
                int cnt=0;
                for (int i=0;i<a.length();i++)
                {
                    if (a[i]!=s[i]) cnt++;
                }
                if (cnt<=1) can=1;
            }
            if (can==1) puts("YES");
            else puts("NO");
        }
        else
        {
            int can=0;
            for (int i=0;i<b.length();i++)
            {
                for (int j=i+1;j<b.length();j++)
                {
                    int cnt=0;
                    for (int k=0;k<b.length();k++)
                    {
                        if (k==i || k==j) continue;
                        if (a[k]!=b[k]) cnt++;
                    }
                    if (cnt==0) can=1;
                }
            }
            for (int i=0;i<b.length();i++)
            {
                for (int j=0;j<b.length();j++)
                {
                    string s="";
                    string t="";
                    for (int k=0;k<b.length();k++)
                    {
                        if (k!=i) s.push_back(a[k]);
                    }
                    for (int k=0;k<b.length();k++)
                    {
                        if (k!=j) t.push_back(b[k]);
                    }
                    if (s==t) can=1;
                }
            }
            if (can==1) puts("YES");
            else puts("NO");
        }
    }
    return 0;
}