首页 > 试题广场 >

病毒检测

[编程题]病毒检测
  • 热度指数:3065 时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 32M,其他语言64M
  • 算法知识视频讲解
小明最近在做病毒自动检测,他发现,在某些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
首先来回顾一下题意:
这是一个子串满足问题,输入一个由01字符组成的字符串str,求满足包含k1的连续字串个数。
毫不夸张的说,可能大部分人第一眼就是求出所有子串,再计数所有满足k的子串,但仔细一想,问题没有那么简单!

思路就是,采用滑动窗口的方式,从字符串数组左侧依次向右滑动,直至满足k个1,在下一个数不为1且数组不越界的情况下窗口向右滑动,同时计数。
核心代码如下:
public class Main {
    public static int NumSubString(int k, String str) {
        char[] chars = str.toCharArray();
        int res = 0;
        int len = chars.length;
        for (int i = 0; i < len; i++) {
            if (chars[i] == '1') {
                res++;
            }
        }
        if (res < k) {
            return 0;
        }
        res = 0;
        for (int i = 0; i < len; i++) {
            /*滑动索引*/
            int index = i;
            int count = 0;
            while (count < k && index < len) {
                if (chars[index] == '1' && ++count == k) {
                    res++;
                    index++;
                    break;
                }
                index++;
            }
            /*满足k后继续右滑*/
            while (index < len && chars[index] != '1') {
                res++;
                index++;
            }
        }
        return res;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int k = sc.nextInt();
        sc.nextLine();
        String str = sc.next();
        System.out.println(NumSubString(k,str));
    }

编辑于 2020-04-11 23:12:07 回复(6)