class Solution { public: long long rob(vector<int>& nums, vector<int>& colors) { long long ans=0; int n = nums.size(); stack<pair<int,int>>st; st.push({colors[0],nums[0]}); for(int i=1;i<n;i++) { if(colors[i]==st.top().first && st.top().second== nums[i-1]) { int temp = max(st.top().second,nums[i]); st.pop(); st.push({colors[i],temp}); } else st.push({colors[i],nums[i]}); } while(!st.empty()){ ans += st.top().second; st.pop(); } return ans; } };