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);
}
}