0
点赞
收藏
分享

微信扫一扫

【力扣每日一题】最长快乐字符串

力扣每日一题20220207

题目简述

链接:力扣1405. 最长快乐字符串

题解:根据本题的题意,很容易想到了深度优先搜索的算法,即dfs+剪枝,此时的算法时间复杂度应该为O(3^max(a,b,c))但是很遗憾,由于参数a,b,c的范围使得该算法的执行时间较慢,导致最终结果为TLE。
代码为:

class Solution {
public:
    string ans;
    int Max;
    string x;
    int num[3];
    string longestDiverseString(int a, int b, int c) {
        map<char, int> s;
        s['a'] = 1;
        s['b'] = 1;
        s['c'] = 1;
        Max = 0;
        ans = "";
        x = "";
        num[0] = a;
        num[1] = b; num[2] = c;
        getLong(s,-1);
        return ans;
    }

    void getLong(map<char, int> s,int alpha)
    {
        int i;
        if (s['a'] == 3)
        {
            if (Max < x.length())
            {
                Max = x.length()-1;
                ans = x.substr(0,x.length()-1);
            }
            return;
        }
        if (s['b'] == 3)
        {
            if (Max < x.length())
            {
                Max = x.length() - 1;
                ans = x.substr(0, x.length() - 1);
            }
            return;
        }
        if (s['c'] == 3)
        {
            if (Max < x.length())
            {
                Max = x.length() - 1;
                ans = x.substr(0, x.length() - 1);
            }
            return;
        }
        for (i = 0; i < 3; i++)
            if (num[i] != 0)
                break;
        if (i == 3) {
            Max = x.length();
            ans = x;
            return;
        }
        for (int i = 0; i < 3; i++)
        {
            if (num[i] > 0)
            {
                num[i]--;
                if (x.length() != 0 && x[x.length() - 1] - 'a' == i) 
                    s['a' + i]++;
                else if (x.length() != 0) {
                    s['a' + i] = 1;
                }
                x += 'a' + i;
                getLong(s,i);
                x = x.substr(0, x.length() - 1);
                if (x.length() != 0 && x[x.length() - 1] - 'a' == i)
                    s['a' + i]--;
                num[i]++;
            }
        }

    }
};

执行结果为:
在这里插入图片描述
最后发现,这并不是一道dfs,而是一道贪心!!!
可以先看本题的题目条件:

  1. 每个字母的连续个数不能超过3个;
  2. 字母的个数有限

而最终所需要得到的结果则是最长的字符串,因此很容易可以想到贪心策略:

代码为:

class Solution {
public:
    class mycomp2 {
    public:
        bool operator() (pair<char, int> a, pair<char, int> b) {
            return (a.second > b.second);
        }
    };
    string longestDiverseString(int a, int b, int c) {
        string ans;
        vector<pair<char, int>> s = { {'a',a},{'b',b},{'c',c} };
        while (1) {
            bool flag = false;
            sort(s.begin(), s.begin()+s.size(), mycomp2());
            for (vector<pair<char, int>>::iterator it = s.begin(); it != s.end(); it++) {
                int m = ans.size();
                char ch = (*it).first;
                int freq = (*it).second;
                if (freq <= 0)
                    break;
                if (m >= 2 && ans[m - 2] == ch && ans[m - 1] == ch)
                    continue;
                ans.push_back(ch);
                flag = 1;
                (*it).second--;
                break;
            }
            if (!flag)
                break;
        }
        return ans;
    }    
};

最终运行结果为:
在这里插入图片描述
采用贪心,则时间复杂度大大降低,主要的时间是在排序上,若采用sort函数,则时间复杂度为O((a+b+c)×ClogC)。

总览这道题,发现采用暴力算法并没有很好的完成这道题,很容易TLE;还是需要多多锻炼贪心或者dp的逻辑思维哇!

举报

相关推荐

0 条评论