数据范围小,直接暴力枚举所有子字符串,统计每个子字符串的0和1的数量就行了。
1 class Solution { 2 public: 3 int countKConstraintSubstrings(string s, int k) { 4 int res=0; 5 for(int i=0;i<s.size();i++){ 6 for(int j=i;j<s.size();j++){ 7 int cnt[2]={0}; 8 for(int k=i;k<=j;k++){ 9 cnt[s[k]-'0']++; 10 } 11 if(cnt[0]<=k||cnt[1]<=k) 12 res++; 13 } 14 } 15 return res; 16 } 17 };