算法提高 项链
时间限制:1.0s 内存限制:512.0MB
问题描述
由 n(1≤n≤100) 个珠子组成的一个项链,珠子有红、蓝、白三种颜色,各种颜色的珠子的安排顺序由键盘输入的字符串任意给定。蓝色用小写字母b表示,红色用小写字母r表示, 白色用小写字母w表示.
假定从项链的某处将其剪断,把它摆成一条直线。先从左端向右收集同色珠子,遇到第一个异色珠子时停止. 收集过程中, 白色是一种特殊颜色, 既可以看成红色也可以看成蓝色。然后再从剩余珠子的右端向左重复上述过程。
例如:对下图一所示的项链, 如果从图一中标记的位置0处剪断, 则按顺时针顺序得到wbbbwwrrbwbrrwb(如图二所示)。这时从左端开始收集可以得到wbbbww, 共6个珠子;然后从剩余珠子右端开始收集得到wb,共2个珠子。这种剪法共可收集到6+2=8个珠子。 如果从图一中标记的位置4处剪断, 则按顺时针顺序得到wwrrbwbrrwbwbbb(如图二所示)。这时从左端收集可以得到wwrr,共4个珠子;然后从剩余珠子右端收集可以得到wbwbbb,共6个珠子。这种剪法共可收集到4+6=10个珠子。
要求: 在项链中选择合适的剪断位置, 使得从左右两端收集到的珠子数目之和最大,输出收集到的珠子数的最大值M。
输入格式
由小写字母b,r,w组成的字符串。此字符串记录了一个首尾相接的项链从某处断开后,按顺时针顺序得到的珠子的直线排列。
输出格式
收集到的珠子数的最大值 M
扫描二维码关注公众号,回复:
9134619 查看本文章
样例输入
wbbbwwrrbwbrrwb
样例输出
10
#include <iostream>
using namespace std;
int main()
{
string necklace;
cin >> necklace;
int len = (int)necklace.length();
int M = 0;
for (int i = 0; i < len; ++i)
{
int collected = 1;
char color = necklace[0];
int idx = 0;
for (; idx < len - 1; ++idx)
{
if (color == 'w')
{
color = necklace[idx + 1];
collected++;
}
else if (color == necklace[idx + 1] || necklace[idx + 1] == 'w')
collected++;
else
break;
}
if (len - 1 > idx)
{
color = necklace[len - 1];
collected++;
for (int j = len - 2; j > idx; --j)
{
if (color == 'w')
{
color = necklace[j];
collected++;
}
else if (color == necklace[j] || necklace[j] == 'w')
collected++;
else
break;
}
}
if (collected > M)
M = collected;
necklace = necklace.substr(1) + necklace[0];
}
cout << M;
return 0;
}