【LeetCode】 198. 打家劫舍
阿新 • • 發佈:2018-11-08
- 題目
你是一個專業的小偷,計劃偷竊沿街的房屋。每間房內都藏有一定的現金,影響你偷竊的唯一制約因素就是相鄰的房屋裝有相互連通的防盜系統,如果兩間相鄰的房屋在同一晚上被小偷闖入,系統會自動報警。
給定一個代表每個房屋存放金額的非負整數陣列,計算你在不觸動警報裝置的情況下,能夠偷竊到的最高金額。
示例 1:
輸入: [1,2,3,1]
輸出: 4
解釋: 偷竊 1 號房屋 (金額 = 1) ,然後偷竊 3 號房屋 (金額 = 3)。
偷竊到的最高金額 = 1 + 3 = 4 。
示例 2:
輸入: [2,7,9,3,1]
輸出: 12
解釋: 偷竊 1 號房屋 (金額 = 2), 偷竊 3 號房屋 (金額 = 9),接著偷竊 5 號房屋 (金額 = 1)。
偷竊到的最高金額 = 2 + 9 + 1 = 12 。
-
思路
當只有一家的時候,返回這家金額;
當有兩家的時候,返回大的那一家;
把每次最優的結果(金額之和)存放在陣列res中;
第i家偷不偷取決於res陣列中i-2項與當前金額之和與res陣列i-1項的大小;
從而每一步都為最優結果 -
程式碼
int rob(vector<int>& nums)
{
int len=nums.size();
if(len==0)
return 0;
if(len==1)
return nums[0];
if(len==2)
return max(nums[0],nums[1]);
int res[len];
res[0]=nums[0];
res[1]=max(nums[0],nums[1]);
for(int i=2;i<len;i++)
{
res[i]=max(nums[i]+res[i-2],res[i-1]);
}
return res[len-1];
}
int main()
{
int a1[]={2,7,9,3,1};
vector<int> g(a1,a1+5);
cout<< rob(g);
return 0;
}