题目来源: Codility
基准时间限制:1 秒 空间限制:131072 KB 分值: 40 难度:4级算法题
51nod 1391:01串-LMLPHP 收藏
51nod 1391:01串-LMLPHP 关注
给定一个01串S,求出它的一个尽可能长的子串S[i..j],满足存在一个位置i<=x <=j, S[i..x]中0比1多,而S[x + 1..j]中1比0多。求满足条件的最长子串长度。
Input
一行包含一个只由0和1构成的字符串S。 S的长度不超过1000000。
Output
一行包含一个整数,表示满足要求的最长子串的长度。
Input示例
10
Output示例
0

hash啊。记录出现值的第一位置,之后如果出现比这个值小于1的数,就说明这个位置到此处的位置中0比1多。

代码:

#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <string>
#include <cstring>
#pragma warning(disable:4996)
using namespace std; #define maxn 1000003 int val[maxn];
int hash_val[maxn];
int le[maxn];
int ri[maxn]; char test[maxn]; int main()
{
//freopen("i.txt", "r", stdin);
//freopen("o.txt", "w", stdout); int i, res, cur, len; scanf("%s", test);
len = strlen(test); for (i = 0; i < len; i++)
{
if (test[i] == '0')
{
val[i] = -1;
}
else
{
val[i] = 1;
}
} memset(hash_val, -1, sizeof(hash_val));
cur = 0;
for (i = 0; i < len; i++)
{
cur = cur + val[i];
if (cur < 0)
{
le[i] = i + 1;
}
else
{
if (hash_val[cur + 1] != -1)
{
le[i] = i - hash_val[cur + 1];
}
else
{
hash_val[cur] = i;
le[i] = 0;
}
}
} memset(hash_val, -1, sizeof(hash_val));
cur = 0;
for (i = len - 1; i >= 0; i--)
{
cur = cur + val[i];
if (cur > 0)
{
ri[i] = len - i;
}
else
{
if (hash_val[-(cur - 1)] != -1)
{
ri[i] = hash_val[-(cur - 1)] - i;
}
else
{
hash_val[-(cur)] = i;
ri[i] = 0;
}
}
}
res = 0;
for (i = 0; i < len; i++)
{
if (le[i] > 0 && ri[i + 1] > 0)
res = max(res, le[i] + ri[i + 1]);
}
printf("%d\n", res);
//system("pause");
return 0;
}

版权声明:本文为博主原创文章,未经博主允许不得转载。

05-11 17:34