CF 1624 E Masha-forgetful 题解

题面

给出一个 \(n\) 个长为 \(m\) 的字符串集和一个同样长度的字符串 \(S\)，用集合中的字符串长度至少为 \(2\) 的子串拼出 \(S\)，输出任意解。同一个字符可以重复使用，多组数据。

题解

注意到不同子串长度可以用长度为 \(2\) 和 \(3\) 的字符串组合出来，那么在一个长度为 \(2\) 和 \(3\) 的子串表里记录子串的编号和串内位置递推 + backtracing 即可。

拓展

递推都出来了，不计个数？其他条件不变，用集合中的字符串长度至少为 \(2\) 的子串拼出 \(S\) 的方案数。

Code

#include <cstdio>
#include <utility>
#include <stack>
#include <cstring>
const int MAXN=1e3+5;
const int MAXD=9+5;
int n,m,t;
std::pair<int, int> jump2[MAXD][MAXD],jump3[MAXD][MAXD][MAXD];
char in[MAXN][MAXN];
std::pair<int, std::pair<int, int>> src[MAXN];
static inline int id(int i, int j) {
    return in[i][j]-'0';
}
int main() {
    scanf("%d",&t);
    while(t--){
        memset(jump2, 0, sizeof(jump2));
        memset(jump3, 0, sizeof(jump3));
        memset(src, 0, sizeof(src));
        scanf("%d %d",&n,&m);
        for(int i=1;i<=n;i++)
            scanf("%s", in[i]+1);
        scanf("%s",in[0]+1);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++) {
                if (j+1<=m) {
                    jump2[id(i,j)][id(i,j+1)]=std::make_pair(i,j);
                    if (j+2<=m) {
                        jump3[id(i,j)][id(i,j+1)][id(i,j+2)]=std::make_pair(i,j);
                    }
                }
            }
        for(int i=1;i<=m;i++) {
            std::pair<int, int> pair;
            if(i==1 || src[i].first) {
                if(i+1<=m && (pair = jump2[id(0,i)][id(0,i+1)]).first) {
                    src[i+2]=std::make_pair(2, pair);
                    if (i+2<=m && (pair = jump3[id(0,i)][id(0,i+1)][id(0,i+2)]).first) {
                        src[i+3]=std::make_pair(3, pair);
                    }
                }
            }
        }
        if(src[m+1].first) {
            std::stack<std::pair<int, std::pair<int, int>>> sta;
            int cur = m+1;
            while (cur > 1) {
                sta.push(src[cur]);
                cur-=src[cur].first;
            }
            printf("%d\n",sta.size());
            while(!sta.empty()) {
                auto u = sta.top();
                printf("%d %d %d\n", u.second.second, u.second.second+u.first-1, u.second.first);
                sta.pop();
            }
        } else {
            printf("-1\n");
        }
    }
    return 0;
}