题目
如果字符串中不含有任何 ‘aaa’,‘bbb’ 或 ‘ccc’ 这样的字符串作为子串,那么该字符串就是一个「快乐字符串」。
给你三个整数 a,b ,c,请你返回 任意一个 满足下列全部条件的字符串 s:
- s 是一个尽可能长的快乐字符串。
- s 中 最多 有a 个字母 ‘a’、b 个字母 ‘b’、c 个字母 ‘c’ 。
- s 中只含有 ‘a’、‘b’ 、‘c’ 三种字母。
如果不存在这样的字符串 s ,请返回一个空字符串 “”。
示例 1:
输入:a = 1, b = 1, c = 7 输出:"ccaccbcc" 解释:"ccbccacc" 也是一种正确答案。
示例 2:
输入:a = 2, b = 2, c = 1 输出:"aabbc"
示例 3:
输入:a = 7, b = 1, c = 0 输出:"aabaa" 解释:这是该测试用例的唯一正确答案。
解题
方法一:贪心+最大堆
最大堆,每次取数量最多(贪心)的字符。
class Solution { public: string longestDiverseString(int a, int b, int c) { struct cmp{ bool operator()(pair<char,int>&a,pair<char,int>&b){ return a.second<b.second; } }; priority_queue<pair<char,int>,vector<pair<char,int>>,cmp> q; if(a>0) q.push({'a',a}); if(b>0) q.push({'b',b}); if(c>0) q.push({'c',c}); string res; while(!q.empty()){ auto cur=q.top(); q.pop(); int n=res.size(); if(n>=2&&res[n-1]==cur.first&&res[n-2]==cur.first){ if(q.empty()) break; auto next=q.top(); q.pop(); res.push_back(next.first); if(--next.second!=0) q.push(next); q.push(cur); } else{ res.push_back(cur.first); if(--cur.second!=0) q.push(cur); } } return res; } };