蓝桥杯2023年第十四届省赛A组-更小的数

题目描述

小蓝有一个长度均为 n 且仅由数字字符 0 ∼ 9 组成的字符串，下标从 0 到 n − 1，你可以将其视作是一个具有 n 位的十进制数字 num，小蓝可以从 num 中选出一段连续的子串并将子串进行反转，最多反转一次。小蓝想要将选出的子串进行反转后再放入原位置处得到的新的数字 num_new 满足条件 num_new < num，请你帮他计算下一共有多少种不同的子串选择方案，只要两个子串在 num 中的位置不完全相同我们就视作是不同的方案。

注意，我们允许前导零的存在，即数字的最高位可以是 0 ，这是合法的。

输入格式

输入一行包含一个长度为 n 的字符串表示 num（仅包含数字字符 0 ∼ 9），从左至右下标依次为 0 ∼ n − 1。

输出格式

输出一行包含一个整数表示答案。

样例输入

210102

样例输出

8

提示

一共有 8 种不同的方案：

1）所选择的子串下标为 0 ∼ 1 ，反转后的 num_new = 120102 < 210102 ；

2）所选择的子串下标为 0 ∼ 2 ，反转后的 num_new = 012102 < 210102 ；

3）所选择的子串下标为 0 ∼ 3 ，反转后的 num_new = 101202 < 210102 ；

4）所选择的子串下标为 0 ∼ 4 ，反转后的 num_new = 010122 < 210102 ；

5）所选择的子串下标为 0 ∼ 5 ，反转后的 num_new = 201012 < 210102 ；

6）所选择的子串下标为 1 ∼ 2 ，反转后的 num_new = 201102 < 210102 ；

7）所选择的子串下标为 1 ∼ 4 ，反转后的 num_new = 201012 < 210102 ；

8）所选择的子串下标为 3 ∼ 4 ，反转后的 num_new = 210012 < 210102 ；

对于 20% 的评测用例，1 ≤ n ≤ 100 ；

对于 40% 的评测用例，1 ≤ n ≤ 1000 ；

对于所有评测用例，1 ≤ n ≤ 5000 。

错误解法

暴力遍历字符串，若 s[i] > s[j] 则满足条件，则答案的个数加 1。 若s[i] = s[j]，则将子串区间 [i, j] 缩小为 [i + 1, j - 1]，再判断 s[i + 1] 和 s[j - 1] 的大小关系。

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;

int main()
{
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	string s;
	cin >> s;
	int ans = 0;
	for(int i = 0; i < s.size() - 1; i++)
	{
		for(int j = i + 1; j < s.size(); j++)
		{
			if(s[i] > s[j])
			{
				ans++;
			}
			else if(s[i] == s[j])
			{
				int u = i, v = j;
				while(u < v)
				{
					u++, v--;
					if(s[u] > s[v])
					{
						ans++;
						break;
					}
					else if(s[u] < s[v])
					{
						break;
					}
				}
			}
		}
	}
	cout << ans << endl;
	return 0;
}

时间复杂度最坏为 O(n^3)（所有数都相同），n 的范围是 5000，显然会超时。

整体思路

考虑区间动态规划，dp[i][j]​ 表示从 i 到 j 反转后是否满足要求，满足要求赋值为 true，否则赋值为 false。

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;

const int N = 5e3 + 5;
bool dp[N][N];

int main()
{
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	string s;
	cin >> s;
	int n = s.size();
	for(int i = 0; i < n - 1; i++)
	{
		dp[i][i + 1] = (s[i] > s[i + 1]);
	}
	for(int len = 2; len <= n; len++)
	{
		for(int i = 0; i < n - len; i++)
		{
			int j = i + len;
			if(s[i] == s[j])
			{
				dp[i][j] = dp[i + 1][j - 1];
			}
			else if(s[i] > s[j])
			{
				dp[i][j] = 1;
			}
		}
	}
	int ans = 0;
	for(int i = 0; i < n - 1; i++)
	{
		for(int j = i + 1; j < n; j++)
		{
			ans += dp[i][j];
		}
	}
	cout << ans << endl;
	return 0;
}

具体步骤

寻找状态转移方程。如果 s[i] > s[j]​，那么 dp[i][j] = true。如果 s[i] = s[j]​，那么 dp[i][j] = dp[i + 1][j - 1]​。整体时间复杂度为 O(n^2)。