題目 1501: [藍橋杯][演算法提高VIP]分蘋果

題目描述

小朋友排成一排，老師給他們分蘋果。
小朋友從左到右標號1…N。有M個老師，每次第i個老師會給第Li個到第Ri個，一共Ri-Li+1個小朋友每人發Ci個蘋果。
最後老師想知道每個小朋友有多少蘋果。

資料規模和約定
100%的資料，N、M≤100 000，1≤Li≤Ri≤N，0≤Ci≤100。

輸入
第一行兩個整數N、M，表示小朋友個數和老師個數。
接下來M行，每行三個整數Li、Ri、Ci，意義如題目表述。
輸出
一行N個數，第i個數表示第i個小朋友手上的水果。
樣例輸入
5 3
1 2 1
2 3 2
2 5 3
樣例輸出
1 6 5 3 3

解題思路:

這道題的解題方式有很多，但錯誤方式也不少，最大最多的問題就是時間超限問題.

這道題的思路就是把每個老師要發的蘋果發給對應的孩子即可.那麼最通用的思路就是迴圈挨個增加，但是這個會時間超限問題.
那麼我們能不能換一個思路，既然一個個去增加不行，那我能不能去提前去把他在那個區間的增量計算出來，然後去從第一個通過+=的方式得到。這樣描述有點兒不合理的，所以我們通過影像來了解一下，這裡我使用上賣弄的樣例進行描述

所以我們可以定義個陣列，就判斷它每一個相對於前一個增加了幾個或減少了幾個，最後通過+=解決即可,具體看程式碼

程式碼如下:

方法一: 這個思路沒有問題,但是時間會超限

import java.util.Scanner;

public class Cyyw1501 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int m = sc.nextInt();
		int[][] arr = new int[m+1][3];
		int [] num = new int[n+1];
		for (int i = 1; i < arr.length; i++) {
			for (int j = 0; j < arr[i].length; j++) {
				arr[i][j] = sc.nextInt();
			}
			for (int j = arr[i][0]; j <= arr[i][1]; j++) {
				num[j]+= arr[i][2];
			}
		}
		
		for (int i = 1; i < num.length; i++) {
			System.out.print(num[i]+" ");
		}
	}
}

方法二 :

import java.util.Scanner;

public class Cyyw1501 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int m = sc.nextInt();
		int[] arr = new int[n];
		for (int i = 0; i < m; i++) {
			int li = sc.nextInt();
			int ri = sc.nextInt();
			int ci = sc.nextInt();
			arr[li-1] += ci; 
			if (ri < n) 
				arr[ri] -= ci;
		}
		
		int num = 0;
		for (int i = 0; i < n; i++) {
			num += arr[i];
			System.out.print(num+" ");
		}
		
		sc.close();
	}
}

執行例項: