题解/算法 {3186. 施咒的最大总伤害}

最新推荐文章于 2024-09-27 14:28:55 发布

「已注销」

最新推荐文章于 2024-09-27 14:28:55 发布

阅读量182

点赞数 1

分类专栏：题解文章标签：算法深度优先

本文链接：https://blog.csdn.net/qq_66485519/article/details/139785483

版权

题解专栏收录该内容

298 篇文章 0 订阅

订阅专栏

题解/算法 {3186. 施咒的最大总伤害}

@LINK: https://leetcode.cn/problems/maximum-total-damage-with-spell-casting/;

这个题啊他卡时间…

很容易想到DP求解, 用map<int,int> Cont来存储然后配合DFS记忆化, 求最大值, 普普通通的DP题

long long maximumTotalDamage(vector<int>& A) {
    map<int,int> Cont;
    for( auto i : A){ Cont[i] ++;}

    unordered_map<int,long long> REC;
    auto Dfs = [&]( auto _dfs, int _a)->long long{
        if( REC.count( _a) > 0){
            return REC[_a];
        }

        long long ANS = 0;
        {
            auto it = Cont.lower_bound( _a+1);
            if( it != Cont.end()){
                ANS = std::max( ANS, _dfs( _dfs, it->first));
            }
        }
        {
            long long sum = _a *1LL* Cont[_a];
            auto it = Cont.lower_bound( _a+3);
            if( it != Cont.end()){
                sum += _dfs( _dfs, it->first);
            }
            ANS = std::max( ANS, sum);
        }
        REC[ _a] = ANS;
        return REC[_a];
    };
    return Dfs( Dfs, Cont.begin()->first);
}

可是他超时了…

分析下时间, DP状态有1e5个, 然后注意不仅lower_bound 还有REC[] 他俩都是log级别的, 按理说没事, 可力扣他的测试数据是算在同一个程序的时间总之他时间限制太严格…

优化吧…

我们把这个lower_bound优化掉就需要把之前 DP状态里是存的数值即map里的元素, 把他变成下标[0, 1e5);
借助vector的一个构造函数 vector( mp.begin(), mp.end()) 他存的是pair即键值对;

于是我们找> cur+2的元素, 就直接通过下标判断要么是ind+1/ ind+2;

别忘了REC[]下标操作, 之前他是map 现在变成数组;

long long maximumTotalDamage(vector<int>& A) {
    vector< pair<int,int> > C;
    {
        map<int,int> Cont;
        for( auto i : A){ Cont[i] ++;}
        C = vector< pair<int,int> >( Cont.begin(), Cont.end());
    }

    int N = C.size();
    static long long REC[int(1e5)+5];
    std::memset( REC, -1, sizeof(REC[0])*N);
    auto Dfs = [&]( auto _dfs, int _ind)->long long{
        if( REC[_ind] == -1){
            REC[ _ind] = 0;
            { // no
                if( _ind + 1 < N){
                    REC[_ind] = max( REC[_ind], _dfs( _dfs, _ind+1));
                }
            }
            { // yes
                long long v = C[_ind].first*1LL*C[_ind].second;
                FOR_( i, _ind+1, N-1){
                    if( C[i].first - C[_ind].first > 2){
                        v += _dfs(_dfs,i);
                        break;
                    }
                }
                REC[ _ind] = std::max( REC[ _ind], v);
            }
        }
        return REC[_ind];
    };
    return Dfs( Dfs, 0);
}

但是把, 他又超时了… 理论上分析他就是1e5啊, 可能是因为递推的缘故, 你的auto _dfs 要写成引用的方式即auto & _dfs 才可以;