トピックの説明
配列 nums とスライディング ウィンドウ サイズ k を指定して、すべてのスライディング ウィンドウの最大値を見つけます。
例:
入力: 数値 = [1,3,-1,-3,5,3,6,7]、および k = 3
出力: [3,3,5,5,6,7]
説明:
スライディングウィンドウの最大位置
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5 1
3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7
モノトニック キュー ソリューション
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
deque<int> window_dq;
vector<int> ans;
for(int i=0;i<k;++i){
if(window_dq.empty()||window_dq.back()>=nums[i]){
window_dq.push_back(nums[i]);
}else{
while(!window_dq.empty()&&window_dq.back()<nums[i]){
window_dq.pop_back();
}
window_dq.push_back(nums[i]);
}
}
if(k!=0)
ans.push_back(window_dq.front());
for(int i=k;i<nums.size();++i){
int window_left=i-k;
if(window_dq.front()==nums[window_left])window_dq.pop_front();
while(!window_dq.empty()&&window_dq.back()<nums[i]){
window_dq.pop_back();
}
window_dq.push_back(nums[i]);
ans.push_back(window_dq.front());
}
return ans;
}
};
multiset 解法
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
multiset<int> win;
vector<int> ans;
if(k==0)return ans;
for(int i=0;i<k;++i){
win.insert(nums[i]);
}
ans.push_back(*win.rbegin());
for(int r=k;r<nums.size();++r){
int l=r-k;
win.erase(win.find(nums[l]));
win.insert(nums[r]);
ans.push_back(*win.rbegin());
}
return ans;
}
};
ブロックDP
配列を左から右に k 個ずつグループ化すると、nums
[i] から nums[i+k−1] までの最大値は 2 つの場合に分けられます。後半はひとつのグループに。各グループは dp を使用して、プレフィックスとサフィックスの最大値を事前に計算します