สมมติว่าเรามีกล่องหลายกล่องที่มีสีต่างกัน สีเหล่านี้แทนด้วยจำนวนบวกต่างกัน เราสามารถสัมผัสได้หลายรอบในการแกะกล่องจนไม่มีกล่องเหลือ ในแต่ละรอบ เราสามารถเลือกกล่องต่อเนื่องที่มีสีเดียวกันได้ (ประกอบด้วย k กล่อง k>=1) แล้วเอาออกและรับ k*k คะแนน ดังนั้นหากอินพุตเป็น − [1,3,2,2,2,4,4,3,1] ผลลัพธ์จะเป็น 21
ค้นหาคะแนนสูงสุดที่คุณจะได้รับ
เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -
- กำหนดฟังก์ชัน Solve() ซึ่งจะใช้กล่องอาร์เรย์ i, j, k, dp อาร์เรย์ 3 มิติหนึ่งชุด
- ถ้า i> j แล้ว −
- คืน 0
- ถ้า dp[i, j, k] ไม่เท่ากับ -1 แล้ว −
- ส่งคืน dp[i, j, k]
- ret :=-inf
- สำหรับการตรวจสอบเงื่อนไข i + 1 <=j และ boxes[i + 1] เหมือนกับ box[i], อัปเดต (เพิ่ม i ขึ้น 1), (เพิ่ม k ขึ้น 1), ไม่ทำอะไรเลย -
- ret :=สูงสุดของ ret และ (k + 1) * (k + 1) + เรียกฟังก์ชัน Solve(boxes, i + 1, j, 0, dp)
- สำหรับการเริ่มต้น x :=i + 1 เมื่อ x <=j อัปเดต (เพิ่ม x ขึ้น 1) ทำ −
- ถ้า box[x] เหมือนกับ box[i] แล้ว −
- ret :=สูงสุดของ ret และแก้ปัญหา ((กล่อง, i + 1, x - 1, 0, dp) + แก้ (กล่อง, x, j, k + 1, dp))
- ถ้า box[x] เหมือนกับ box[i] แล้ว −
- ส่งคืน dp[i, j, k] =ret
- จากวิธีหลัก ให้ทำดังนี้
- n :=ขนาดของกล่อง
- กำหนดอาร์เรย์ 3 มิติ dp ของคำสั่ง (n + 1) x (n + 1) x (n + 1) หนึ่งชุด เติมด้วย -1
- ส่งคืนการแก้(กล่อง 0, n - 1, 0, dp)
ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -
ตัวอย่าง
#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
int solve(vector <int>& boxes, int i, int j, int k, vector < vector < vector <int > > >& dp){
if(i > j) return 0;
if(dp[i][j][k] != -1) return dp[i][j][k];
int ret = INT_MIN;
for(; i + 1 <= j && boxes[i + 1] == boxes[i]; i++, k++);
ret = max(ret, (k + 1) * (k + 1) + solve(boxes, i + 1, j, 0, dp));
for(int x = i + 1; x <= j; x++){
if(boxes[x] == boxes[i]){
ret = max(ret, solve(boxes, i + 1, x - 1, 0, dp) + solve(boxes, x, j, k + 1, dp));
}
}
return dp[i][j][k] = ret;
}
int removeBoxes(vector<int>& boxes) {
int n = boxes.size();
vector < vector < vector <int > > > dp(n + 1, vector < vector <int> > (n + 1, vector <int>(n + 1, -1)));
return solve(boxes, 0, n - 1, 0, dp);
}
};
main(){
Solution ob;
vector<int> v = {1,3,2,2,2,4,4,3,1};
cout << (ob.removeBoxes(v));
} อินพุต
{1,3,2,2,2,4,4,3,1} ผลลัพธ์
21