差分数组工具类:
class Difference{
? ?private int[] diff;
?
? ?public Difference(int[] nums){
? ? ? ?assert nums.length>0;
? ? ? ?diff = new int[nums.length];
? ? ? ?diff[0] = nums[0];
?
? ? ? ?for(int i =1 ;i<nums.length;i++){
? ? ? ? ? ?diff[i]= nums[i]-nums[i-1];
? ? ? }
? }
?
? ?public void increasement(int i,int j, int val){
? ? ? ?diff[i]+=val;
? ? ? ?if(j+1<diff.length){
? ? ? ? ? ?diff[j+1]-=val;
? ? ? }
?
? }
?
? ?public int[] res(){
? ? ? ?int[] res = new int[diff.length];
? ? ? ?res[0]=diff[0];
? ? ? ?for(int i =1 ;i<diff.length;i++){
? ? ? ? ? ?res[i] = res[i-1]+diff[i];
? ? ? }
? ? ? ?return res;
?
? }
差分数组的应用:
Leetcode1109:航班预定统计
这里有 n 个航班,它们分别从 1 到 n 进行编号。
有一份航班预订表 bookings ,表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti (包含 firsti 和 lasti )的 每个航班 上预订了 seatsi 个座位。
请你返回一个长度为 n 的数组 answer,里面的元素是每个航班预定的座位总数。
class Solution {
? ?public int[] corpFlightBookings(int[][] bookings, int n) {
? ? ? ?int[] nums = new int[n];
? ? ? ?Difference df = new ?Difference(nums);
? ? ? ?for(int[] booking:bookings){
? ? ? ? ? ?int i =booking[0]-1;
? ? ? ? ? ?int j = booking[1]-1;
? ? ? ? ? ?int val = booking[2];
? ? ? ? ? ?df.increasement(i,j,val);
? ? ? }
?
? ? ? ?return df.res();
? }
?
? ?
}
?
class Difference{
? ?private int[] diff;
?
? ?public Difference(int[] nums){
? ? ? ?assert nums.length>0;
? ? ? ?diff = new int[nums.length];
? ? ? ?diff[0] = nums[0];
?
? ? ? ?for(int i =1 ;i<nums.length;i++){
? ? ? ? ? ?diff[i]= nums[i]-nums[i-1];
? ? ? }
? }
?
? ?public void increasement(int i,int j, int val){
? ? ? ?diff[i]+=val;
? ? ? ?if(j+1<diff.length){
? ? ? ? ? ?diff[j+1]-=val;
? ? ? }
?
? }
?
? ?public int[] res(){
? ? ? ?int[] res = new int[diff.length];
? ? ? ?res[0]=diff[0];
? ? ? ?for(int i =1 ;i<diff.length;i++){
? ? ? ? ? ?res[i] = res[i-1]+diff[i];
? ? ? }
? ? ? ?return res;
?
? }
}
Leetcode1094.拼车
题目:车上最初有 capacity 个空座位。车 只能 向一个方向行驶(也就是说,不允许掉头或改变方向)
给定整数 capacity 和一个数组 trips , trip[i] = [numPassengersi, fromi, toi] 表示第 i 次旅行有 numPassengersi 乘客,接他们和放他们的位置分别是 fromi 和 toi 。这些位置是从汽车的初始位置向东的公里数。
当且仅当你可以在所有给定的行程中接送所有乘客时,返回 true,否则请返回 false。
class Solution {
? ?public boolean carPooling(int[][] trips, int capacity) {
? ? ? ?int[] nums = new int[1001];
? ? ? ?Difference df = new Difference(nums);
?
? ? ? ?for(int[] trip:trips){
? ? ? ? ? ?int val = trip[0];
? ? ? ? ? ?int i = trip[1];
? ? ? ? ? ?int j = trip[2]-1;
? ? ? ? ? ?df.increasement(i,j,val);
?
? ? ? }
?
? ? ? ?int[] res = df.res();
? ? ? ?for(int i =0 ; i<res.length;i++){
? ? ? ? ? ?if(res[i]>capacity){
? ? ? ? ? ? ? ?return false;
? ? ? ? ? }
? ? ? }
? ? ? ?return true;
?
? ? ? ?
?
? }
?
? ?class Difference{
?
? ? ? ?private int[] diff ;
?
? ? ? ?public Difference(int[] nums){
? ? ? ? ? ?assert nums.length>0;
? ? ? ? ? ?diff = new int[nums.length];
? ? ? ? ? ?diff[0]=nums[0];
? ? ? ? ? ?for(int i =1;i<nums.length;i++){
? ? ? ? ? ? ? ?diff[i] = nums[i] - nums[i-1];
? ? ? ? ? }
?
? ? ? }
?
? ? ? ?public void increasement(int i,int j ,int val){
? ? ? ? ? ?diff[i]+=val;
? ? ? ? ? ?if(j+1<diff.length){
? ? ? ? ? ? ? ?diff[j+1]-=val;
? ? ? ? ? }
? ? ? }
?
? ? ? ?public int[] res(){
? ? ? ? ? ?int[] res=new int[diff.length];
? ? ? ? ? ?res[0]=diff[0];
? ? ? ? ? ?for(int i =1;i<diff.length;i++){
? ? ? ? ? ? ? ?res[i] = res[i-1]+diff[i];
? ? ? ? ? }
? ? ? ? ? ?return res;
?
? ? ? }
? }
}
|