C市现在要转移一批罪犯到D市,C市有n名罪犯,按照入狱时间有顺序,另外每个罪犯有一个罪行值,值越大罪越重。现在为了方便管理,市长决定转移入狱时间连续的c名犯人,同时要求转移犯人的罪行值之和不超过t,问有多少种选择的方式?
输入描述:第一行数据三个整数:n,t,c(1≤n≤2e5,0≤t≤1e9,1≤c≤n),第二行按入狱时间给出每个犯人的罪行值ai(0≤ai≤1e9)。
输入例子:
3 100 2
1 2 3
输出:
2
思路1:利用滑动窗口的性质来做建立一个c大小的滑动窗口,从开始一直滑到最后,每次进行和t判断,如果符合就加一,不符合就往后面滑动。时间复杂度O(n).
代码如下:
public static void main(String args[]){
Scanner in=new Scanner(System.in);
while(in.hasNext()){
int n = in.nextInt();
int t = in.nextInt();
int c = in.nextInt();
int[] a = new int[n];
for(int i = 0;i<n;i++){
a[i] = in.nextInt();
}
int count = 0;
int tempt = 0;
//建立起始滑动窗口
for(int i = 0;i<c;i++){
tempt += a[i];
}
if(tempt<=t)
count++;
for(int i = c;i<a.length;i++){
tempt = tempt+a[i]-a[i-c];
if(tempt<=t)
count++;
}
System.out.println(count);
}
}
思路2:暴力,时间复杂度O(cn).
public class Main {
public static void main(String[] argv){
List<Integer> criminalValue = new ArrayList<Integer>();
int criminalNum = -1;
int maxCrime = -1;
int selectNum = -1;
int count = 0;
Scanner sc = new Scanner(System.in);
if(sc.hasNext()){
//罪犯人数
criminalNum = sc.nextInt();
//最大罪行值
maxCrime = sc.nextInt();
//连续人数
selectNum = sc.nextInt();
//每个罪犯的罪行值
for(int i = 0; i < criminalNum; i++){
criminalValue.add(sc.nextInt());
}
}
sc.close();
for(int i = 0; i <= criminalNum - selectNum; i++){
int tmp = 0;
for(int j = i; j < i + selectNum; j++){
tmp += criminalValue.get(j);
}
if(tmp <= maxCrime) count++;
}
System.out.println(count);
}
}