洛谷 P1019 单词接龙

题目描述

单词接龙是一个与我们经常玩的成语接龙相类似的游戏，现在我们已知一组单词，且给定一个开头的字母，要求出以这个字母开头的最长的“龙”（每个单词都最多在“龙”中出现两次），在两个单词相连时，其重合部分合为一部分，例如 beast和astonish，如果接成一条龙则变为beastonish，另外相邻的两部分不能存在包含关系，例如at 和 atide 间不能相连。

输入输出格式

输入格式：

输入的第一行为一个单独的整数n (n<=20)表示单词数，以下n 行每行有一个单词，输入的最后一行为一个单个字符，表示“龙”开头的字母。你可以假定以此字母开头的“龙”一定存在.

输出格式：

只需输出以此字母开头的最长的“龙”的长度

输入输出样例

输入样例#1：

5
at
touch
cheat
choose
tact
a

输出样例#1：

23 （连成的“龙”为atoucheatactactouchoose）

说明

NOIp2000提高组第三题

分析

其实直接记忆话搜索就好了，但是总是连续选两个同样的也可能对答案产生贡献。

代码

#include <bits/stdc++.h>

#define N 105

int ans=0;
int n;
int len[N],use[N];
std::string s[N];

bool find(int l,int x,int y)
{
    for (int i = l; i <= len[x] - 1; i++)
    {
        int j = i - l;
        if (s[x][i] != s[y][j])
            return false;
    }
    return true;
}

int getSame(int x,int y)
{
    for (int i = len[x] - 1; i >= 0; i--)
    {
        int same = len[x] - i;
        if(find(i,x,y))
            return same;
    }
    return 0;
}

void dfs(int l,int dep)
{
    if (l > ans)
        ans=l;
    use[dep] ++;
    for (int i = 1; i <= n; i++)
    {
        int same = getSame(dep,i);
        if (same > 0 && use[i] < 2)
            if (same  != len[i])
                dfs(l + len[i] - same,i);
    }
    use[dep] --;
}

int main()
{
    scanf("%d",&n);
    for (int i = 1; i <= n; i++)
        std::cin>>s[i],len[i] = s[i].size();

    char start;
    std::cin>>start;
    for (int i = 1; i <= n; i++)
        if (s[i][0] == start)
            dfs(len[i],i);

    printf("%d\n",ans);
}