CodeForces 54 D.Writing a Song（dfs）

Description

要求用前$K$个小写字母构造一个长度为$n$的字符串，使得给出的一个长度为$m$的字符串与该字符串的匹配状态为所给的一个长度为$n-m+1$的$01$串，该$01$串第$i$位为$1$表示构造的字符串第$i$位开始的一个长度为$m$的子串与所给字符串一致，否则不一致，问是否有解

Input

第一行两个整数$n,K$表示字符串长度和字母表大小，之后输入一长度为$m$的字符串和一个长度为$n-m+1$的$01$串

$(1\le n\le 100,1\le K\le 26)$

Output

如果有解则输出任一解，否则输出$No\ solution$

Sample Input

5 2
aba
101

Sample Output

ababa

Solution

先把必须匹配的部分确定，如果有冲突则无解，对于不确定的位置直接从左到右暴搜即可，每固定一位就往前判断是否与所给字符串匹配，如果匹配则不合法否则继续深搜

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f;
int n,m,K,vis[105];
char s[105],a[105],ans[105];
bool check(int pos)
{
    if(pos<m-1)return 1;
    for(int i=0;i<m;i++)
        if(ans[pos-i]!=s[m-1-i])return 1;
    return 0;
}
bool dfs(int pos)
{
    if(pos==n)return 1;
    if(ans[pos]==0)
    {
        for(int i=0;i<K;i++)
        {
            ans[pos]=i+'a';
            if(check(pos)&&dfs(pos+1))return 1;
        }
        return 0;
    }
    else
    {
        if(vis[pos])return dfs(pos+1);
        if(check(pos)&&dfs(pos+1))return 1;
        return 0;
    }
}
int main()
{
    scanf("%d%d%s%s",&n,&K,s,a);
    m=strlen(s);
    memset(ans,0,sizeof(ans));
    int flag=1;
    for(int i=0;i<n-m+1;i++)
        if(a[i]=='1')
        {
            for(int j=0;j<m;j++)
                if(ans[i+j]==0)ans[i+j]=s[j];
                else if(ans[i+j]!=s[j])flag=0;
            if(!flag)break;
            vis[i+m-1]=1;
        }
    if(flag)
    {
        ans[n]='\0'; 
        if(dfs(0))printf("%s\n",ans);
        else printf("No solution\n");
    }
    else printf("No solution\n");
    return 0;
}