关于LCS输出最长序列的问题

题目来源于atcoder
解题思路参考这篇文章

问题描述


题意：其实就是LCS的一个变形，在LCS的基础上，需要输出这个最长的公共子序列（得先完全理解了LCS的做法后才能来做这道题）

这个我本来想的是用一个二维string数组记录每个状态的最长公共子序列是什么，但是怎么写都是超内存的，所以我直接去找题解了O(∩_∩)O
看了下网上大佬的思路：用了一个flag数组来代表每个状态是从哪个状态转移过来的。
对于当前状态f[i][j]，一共可能从三种状态转移过来（状态转移方程的推导这里我就不写了，已经写了好多遍了）：
1）从f[i-1][j-1]转移过来： 前提条件a[i]==a[j]并且f[i-1][j-1]+1是最大值，将flag[i][j]=0
2）从f[i-1][j]转移过来：前提条件f[i-1][j]是最大值，将flag[i][j]=-1
3）从f[i][j-1]转移过来：前提条件f[i][j-1]是最大值，将flag[i][j]=1

然后完成一遍状态转移，将每一个f[i][j]和flag[i][j]的值更新完毕

接下来分别从两个序列的最后一个字符串开始，根据flag[i][j]的值进行向前搜索：
如果搜索到了flag[i][j]=0的时候，在回溯时需要将这个点的字符打印出来，并递归向前搜索flag[i-1][j-1]
如果flag[i][j]=-1，向前搜索flag[i-1][j]。
如果flag[i][j]=1，向前搜索flag[i][j-1]。

最后回溯时打印该打印的字符即可。

这个地方有个最难懂的地方就是为什么在回溯时需要输出flag[i][j]=1这个状态对应的字符

首先，flag[i][j]=1的话代表的是在dp过程中，添加这个点后，该序列所有集合里长度最长的公共子序列。
其次，回溯时打印，是为了正向打印，否则输出的就是逆序字符串。

具体的见代码如下

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 3005;

char a[N],b[N];
int f[N][N],flag[N][N];


void LCS()
{
    int lena=strlen(a+1);
    int lenb=strlen(b+1);
    for (int i = 1; i <= lena; i ++ )
    for (int j = 1; j <= lenb; j ++ )
    {
        if(a[i]==b[j])
            f[i][j]=f[i-1][j-1]+1,flag[i][j]=0;
        else if(f[i-1][j]>=f[i][j-1])
            f[i][j]=f[i-1][j],flag[i][j]=-1;
        else
            f[i][j]=f[i][j-1],flag[i][j]=1;
    }
}

void printLCS(int i,int j)
{
    if(i==0 || j==0)//当i和j任意一个为0的时候，搜索完毕，开始回溯
        return;
    if(flag[i][j]==0)
    {
        printLCS(i-1,j-1);//向前回溯
        cout<<a[i];//回溯时输出这个字符
    }
    
    else if(flag[i][j]==-1)
        printLCS(i-1,j);
    
    else if(flag[i][j]==1)
        printLCS(i,j-1);
}

int main()
{
    cin>>(a+1)>>(b+1);
    LCS();
    printLCS(strlen(a+1),strlen(b+1));
    return 0;
}