外观数列是指具有以下特点的整数序列:
d, d1, d111, d113, d11231, d112213111, ...
它从不等于 1 的数字 d 开始,序列的第 n+1 项是对第 n 项的描述。比如第 2 项表示第 1 项有 1 个 d,所以就是 d1;第 2 项是 1 个 d(对应 d1)和 1 个 1(对应 11),所以第 3 项就是 d111。又比如第 4 项是 d113,其描述就是 1 个 d,2 个 1,1 个 3,所以下一项就是 d11231。当然这个定义对 d = 1 也成立。本题要求你推算任意给定数字 d 的外观数列的第 N 项。
输入格式:
输入第一行给出 [0,9] 范围内的一个整数 d、以及一个正整数 N(≤ 40),用空格分隔。
输出格式:
在一行中给出数字 d 的外观数列的第 N 项。
输入样例:
1 8
输出样例:
1123123111
思路:
用pre记录一串连续相同的字符,count记录这串字符的长度,当遇到与pre不同的字符时产生部分结果,pre重新赋值,count置1
注意:循环中若出现str = str + c
之类的字符串连接后再赋值给自身的代码,将导致最后一个测试点超时。注意str = str + c
和str += c
时间复杂度的差别,前者为O(n),n为str的长度,后者为O(1)。
#include <bits/stdc++.h>
using namespace std;
int main()
{
ios::sync_with_stdio(false);
int n;
string str;
cin >> str >> n;
for (int i = 1; i < n; i++)
{
string temp;
char pre = str[0];
int count = 1;
for (int j = 1; j <= str.size(); j++) //str[str.size()]为'\0',一定与pre不同,保证最后一串字符能够记录
{
if (str[j] == pre)
count++;
else
{
temp += pre; //改为temp = temp + pre超时
temp += count + '0'; //同上
count = 1;
pre = str[j];
}
}
str = temp;
}
cout << str;
return 0;
}
代码还可以进一步简化,用额外一层内循环寻找连续相同字符的尾部,即可不记录pre,相同字符的串首尾下标相减即为count
#include <bits/stdc++.h>
using namespace std;
int main()
{
ios::sync_with_stdio(false);
int n;
string str;
cin >> str >> n;
for (int i = 1; i < n; i++)
{
string temp;
for (int j = 0, k; j < str.size(); j = k)
{
for (k = j; k < str.size() && str[k] == str[j]; k++);
temp += str[j]; //改为temp = temp + str[j]超时
temp += k - j + '0'; //同上
}
str = temp;
}
cout << str;
return 0;
}