5118-航班預(yù)訂統(tǒng)計

fsmStudy 發(fā)布于2019-08-16 18:29 / 2424人閱讀

摘要：前言的航班預(yù)訂統(tǒng)計這里有個航班，它們分別從到進(jìn)行編號。此時第行數(shù)據(jù)表示起點是，終點是，座位數(shù)是。此時實現(xiàn)代碼航班預(yù)訂統(tǒng)計處理每一條預(yù)訂記錄給對應(yīng)航班編號增加座位數(shù)

前言

Weekly Contest 144的航班預(yù)訂統(tǒng)計：

這里有 n 個航班，它們分別從 1 到 n 進(jìn)行編號。
我們這兒有一份航班預(yù)訂表，表中第 i 條預(yù)訂記錄 bookings[i] = [i, j, k] 意味著我們在從 i 到 j 的每個航班上預(yù)訂了 k 個座位。
請你返回一個長度為 n 的數(shù)組 answer，按航班編號順序返回每個航班上預(yù)訂的座位數(shù)。
示例：
輸入：bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
輸出：[10,55,45,25,25]
提示：

1 <= bookings.length <= 20000
1 <= bookings[i][0] <= bookings[i][1] <= n <= 20000
1 <= bookings[i][2] <= 10000

解題思路

本題題目的思路其實比較簡答：

讀取出每條預(yù)定記錄bookings[i] = [i, j, k]的起點i，終點j和座位數(shù)k

處于起點i和j之間的result[n]需要增加對應(yīng)的座位數(shù)k，即若i<=n+1<=k（因為n為數(shù)組下標(biāo)索引，所以需要n+1），則result[n]+=k

下面會通過題目的示例分析：

輸入：bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
輸出：[10,55,45,25,25]

運算過程

第1行數(shù)據(jù)[1,2,10]表示起點是1，終點是2，座位數(shù)是10。所以result[0]+=10，result[1]+=10。此時

result[0]=10
result[1]=10
result[2]=0
result[3]=0
result[4]=0

第2行數(shù)據(jù)[2,3,20]表示起點是2，終點是3，座位數(shù)是20。所以result[1]+=20，result[2]+=20。此時

result[0]=10
result[1]=30
result[2]=20
result[3]=0
result[4]=0

第3行數(shù)據(jù)[2,5,25]表示起點是2，終點是5，座位數(shù)是15。所以result[1]+=25，result[2]+=25，result[3]+=25，result[4]+=25。此時

result[0]=10
result[1]=55
result[2]=45
result[3]=25
result[4]=25

實現(xiàn)代碼

    /**
     * 5118. 航班預(yù)訂統(tǒng)計
     * @param bookings
     * @param n
     * @return
     */
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] result = new int[n]; 
        for (int i = 0; i < bookings.length; i++) { // 處理每一條預(yù)訂記錄
            int start = bookings[i][0];
            int end = bookings[i][1];
            int seat = bookings[i][2];
            for (int j = start - 1; j < end; j++) { // 給對應(yīng)航班編號增加座位數(shù)
                result[j] += seat;
            }
        }
        return result;
    }