最近刷题,按照题目的难度顺序刷到了这一题,一开始写的代码都因为超时而没有ac过,经过百度后看了一下别人的思路后感叹自己之前的逻辑是有多么的耗时,下面是我之前的代码和
题目描述:
给定一个字符串 s,计算具有相同数量0和1的非空(连续)子字符串的数量,并且这些子字符串中的所有0和所有1都是组合在一起的。
重复出现的子串要计算它们出现的次数。
示例 1 :
输入: "00110011"输出: 6解释: 有6个子串具有相同数量的连续1和0:“0011”,“01”,“1100”,“10”,“0011” 和 “01”。 请注意,一些重复出现的子串要计算它们出现的次数。 另外,“00110011”不是有效的子串,因为所有的0(和1)没有组合在一起。
示例 2 :
输入: "10101"输出: 4解释: 有4个子串:“10”,“01”,“10”,“01”,它们具有相同数量的连续1和0。
超时代码(C#):
public class Solution { public int CountBinarySubstrings(string s) { string temp; int count = 0; int length = 2; bool sign = false; while (length <= s.Length) { for (int i = 0; i < s.Length - length + 1; i++) { sign = false; temp = s.Substring(i, length); int zeroend = temp.LastIndexOf('0'); int zerostart = temp.IndexOf('0'); int onestart = temp.IndexOf('1'); int oneend = temp.LastIndexOf('1'); if (zerostart == -1 || zeroend == -1 || oneend == -1 || onestart == -1) { sign = true; continue; } for (int j = zerostart + 1; j < zeroend; j++) { if (temp[j] != '0') { sign = true; break; } } for (int j = onestart + 1; j < oneend; j++) { if (temp[j] != '1') { sign = true; break; } } if (!sign) count++; } length += 2; } return count; } }
之前的思路是暴力求解的思路,从字符串的第一位到其Length-字串的长度+1为止,求出0,1的开始和结束索引,然后再用循环在这个区间内判断是否全部为0或1,如果是的话就count++,最后返回count的值就可以了,但是由于时间复杂度较高就会出现超时的情况,那么就更改为下面的算法:
更改后代码:
public class Solution { public int CountBinarySubstrings(string s) { int a = 1, b = 0; char num; bool sign = false; int count=0; num = s[0]; int i = 0; int index = 0; while (i<s.Length-1) { i++; if (s[i] == num && sign == false) { a++; } else if (s[i] == num && sign == true) { b++; } else if (s[i] != num && !sign) { b++; index = i; sign = true; num = s[i]; } else if (s[i] != num && sign) { if (a > b) count += b; else count += a; a = 1; b = 0; i = index; sign = false; } if(i==s.Length-1) { if (a > b) count += b; else count += a; } } return count; } }
更改后算法的思路就是从字符串的开始到末尾遍历,求出连续的1或者0的个数(算法中的a),若后面的数不同于前面的数,先记下当前的索引位置然后再求后面的数相同的个数(算法中的b)直到再次遇到与当前数不同的数或者到末尾为止,然后比较a,b的大小,count+=两个数中的小数,那么count就是所求的答案。