题目1
题目描述
小易觉得高数课太无聊了,决定睡觉。不过他对课上的一些内容挺感兴趣,所以希望老师讲到有趣的地方的时候叫醒他一下。你知道了小易对一堂课每分钟知识点的感兴趣程度,并以分数量化,以及他在这堂课上每分钟是否会睡着,你可以叫醒他一次,这会使得他在接下来的k分钟内保持清醒。你需要选择一种方案最大化小易这堂课听到的知识点分值。
输入描述
第一行n,k, (1 <= n, k <= 105),表示这堂课持续多少分钟,以及叫醒小易一次使他能够保持清醒的时间。
第二行 n个数,a1,a2,…,an(1<=ai<=104),表示小易对每分钟知识点的感兴趣评分。
第三行 n个数,t1,t2,…,tn,表示每分钟小易是否清醒,1表示清醒。
输出描述
小易这堂课听到的知识点的最大兴趣值。
示例
输入
6 3
1 3 5 2 5 4
1 1 0 1 0 0
输出
16
思路1
- 我的思路是统计所有为 1 的 时间的兴趣值,O(N)
- 再次遍历,遇到为 0 的时候,统计 之后的 k 个数,并且跳过为 1 个数字不统计
- 比较并获得最大值
- 这种方法的实质也算是暴力法,更好的方法应该是使用动态规划
代码: AC 90%
import java.util.Scanner;
public class testkeshui {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int k = sc.nextInt(); sc.nextLine();
int[][] map = new int[n][2];
String[] str1 = sc.nextLine().split(" ");
String[] str2 = sc.nextLine().split(" ");
for (int i=0;i<n;i++){
map[i][0] = Integer.valueOf(str1[i]);
map[i][1] = Integer.valueOf(str2[i]);
}
int resultMax = 0;
int curMax = 0;
int wake = 0; //醒着的时间获得的兴趣值
for (int i=0;i<map.length;i++){
if (map[i][1] == 1){
wake += map[i][0];
}
}
for (int i=0;i<map.length;i++){
if (map[i][1] == 0){
if (map[i][1] == 0){
//从 i 开始,往后加上 k 个数
curMax = getSum(map,i,k);
curMax = curMax+wake;
}
resultMax = Math.max(resultMax,curMax);
}
}
resultMax = Math.max(resultMax,curMax);
System.out.println(resultMax);
}
private static int getSum(int[][] map, int i,int k){
int res = 0; //接下来的 k 包含当前吗?
for (