病毒检测笔试题题目解析

Problem
小明最近在做病毒自动检测，他发现，在某些library 的代码段的二进制表示中，如果包含子串并且恰好有k个1，就有可能有潜在的病毒。library的二进制表示可能很大，并且子串可能很多，人工分析不可能，于是他想写个程序来先算算到底有多少个子串满足条件。如果子串内容相同，但是开始或者结束位置不一样，则被认为是不同的子串。

注：子串一定是连续的。例如"010"有6个子串，分别是 "0, “1”, “0”, “01”, “10”, “010”

输入描述:

第一行是一个整数k，表示子串中有k个1就有可能是病毒。其中 0 <= k <= 1 000 000

第二行是一个字符串，就是library的代码部分的二进制表示。字符串长度 <= 1 000 000。并且字符串中只包含"0"或"1".

输出描述:

输出一个整数，所有满足只包含k个1的子串的个数。

示例1

输入

1
1010

输出

6

说明

满足条件的子串有：“1”, “1”, “10”, “01”, “10”, “010”.
示例2

输入

2
01010

输出

4

说明

满足条件的子串有： “101”, “0101”, “1010”, “01010”.

示例3

输入

100
01010

输出

0

题解及思路

思路
动态规划思路，当前状态依赖前k个状态并且需维护当前状态为后面服务，需要k+1空间，但是需要开辟k+2个空间，其中一个空间来消除掉之前状态的值。

可能从不同的角度理解dp数组的描述有区别，调试一遍大概能知道啥意思，本来我是想着k+1个数组，发现更新当前状态的时候的时候会出现问题，k+2刚好能满足

其实这个直接创建一个大小为字符串长度的dp数组会更好理解，for循环过程如下：

        for(char c:s.toCharArray()){
            if(c=='1') num++;
            if(num-k>=0) result+=dp[num-k];
            dp[num]++;
        }

dp[i]记录的是i个1的时候字符串的个数，仔细分析一下会发现dp只依赖前k个状态，不需要创建字符串长度大小的数组，就优化成我代码所示那样子

题解

import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner sc=new Scanner(System.in);
        String l1=sc.nextLine();
        String l2=sc.nextLine();
        int k=Integer.parseInt(l1);
        char[] nums=l2.toCharArray();
        int num=0;
        long ans=0;
        HashMap<Integer,Integer> map=new HashMap<>();
        map.put(0,1);
        for(int i=0;i<nums.length;i++){
            if(nums[i]=='1') num+=1;
            if(map.containsKey(num-k)) ans+=map.get(num-k);
            map.put(num,map.getOrDefault(num,0)+1);
        }
        System.out.println(ans);
    }
}