接龙数列 DP（蓝桥杯）

题目描述

对于一个长度为 K 的整数数列：A1, A2, . . . , AK，我们称之为接龙数列当且仅当 Ai 的首位数字恰好等于 Ai−1 的末位数字 (2 ≤ i ≤ K)。

例如 12, 23, 35, 56, 61, 11 是接龙数列；12, 23, 34, 56 不是接龙数列，因为 56的首位数字不等于 34 的末位数字。所有长度为 1 的整数数列都是接龙数列。

现在给定一个长度为 N 的数列 A1, A2, . . . , AN，请你计算最少从中删除多少个数，可以使剩下的序列是接龙序列？

输入格式

第一行包含一个整数 N。

第二行包含 N 个整数 A1, A2, . . . , AN。

输出格式

一个整数代表答案。

样例输入

5
11 121 22 12 2023

 样例输出

1

提示

删除 22，剩余 11, 121, 12, 2023 是接龙数列。

对于 20% 的数据，1 ≤ N ≤ 20。

对于 50% 的数据，1 ≤ N ≤ 10000。

对于 100% 的数据，1 ≤ N ≤ 105，1 ≤ Ai ≤ 109。所有 Ai 保证不包含前导 0。

暴力做法DFS

对于每个数有选和不选两个情况，通过DFS暴力枚举所有情况

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 10;
int arr[N];
int n, ans;

int get_first(int x)
{
    while (x > 9)
        x /= 10;
    return x;
}

int get_last(int x)
{
    return x % 10;
}

// k当前是第几个数，len当前的长度，last前一个数
void dfs(int k, int len, int last)
{
    if (k == n)
    {
        ans = max(ans, len);
        return;
    }
    if (n - k + len <= ans) // 剩余的个数 + 当前个数 <= 当前求出的最大个数时, 不需要再判断了
        return;
    if (last == -1 || get_last(last) == get_first(arr[k])) // 选
        dfs(k + 1, len + 1, arr[k]);
    dfs(k + 1, len, last); // 不选
}

// DFS(可过部分)
int main()
{
    cin >> n;
    for (int i = 0; i < n; ++i)
        cin >> arr[i];
    dfs(0, 0, -1);
    cout << n - ans << endl;
    return 0;
}

正解DP

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 10, INF = 0x3f3f3f3f;
int arr[N], dp[N][15];
int n;

int get_first(int x)
{
    while (x > 9)
        x /= 10;
    return x;
}

int get_last(int x)
{
    return x % 10;
}

// DP
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);
    cin >> n;
    for (int i = 0; i < n; ++i)
        cin >> arr[i];
    // 以i位置为结尾, 最后一位是j的接龙数列的最小删除数
    // 根据状态表示写出状态转移方程
    for (int i = 1; i < n + 1; ++i)
    {
        for (int j = 0; j < 10; ++j)
        {
            dp[i][j] = dp[i - 1][j] + 1;
            if (j == get_last(arr[i - 1])) // dp表多加了一行，arr下标减一
                dp[i][j] = min(dp[i][j], dp[i - 1][get_first(arr[i - 1])]);
        }
    }
    int ans = INF;
    for (int j = 0; j < 10; ++j)
        ans = min(ans, dp[n][j]);
    cout << ans << endl;
    return 0;
}