【牛客】【百度編程】罪犯轉移 【動態規劃】

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

}


發佈了65 篇原創文章 · 獲贊 26 · 訪問量 23萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章