【训练题5：GarsiaWachs算法】石子合并 | 洛谷 P5569

最新推荐文章于 2023-03-05 17:12:36 发布

溢流眼泪

最新推荐文章于 2023-03-05 17:12:36 发布

阅读量377

点赞数 2

分类专栏：【各类ACM真题】文章标签：算法

本文链接：https://blog.csdn.net/weixin_45775438/article/details/109599441

版权

【各类ACM真题】专栏收录该内容

130 篇文章 22 订阅

订阅专栏

石子合并 | 洛谷 P5569

难度

$4 / 10$
经典题

题目

放着 $N$ 堆石子。每堆石子有 $a_i$ 个石头。
每次选择相邻的两堆石子 $i\ ,\ j$ ，把它合并成一堆新的，并获得 $a_i+a_j$ 分数。
计算出将该 $N$ 堆石子合并成一堆的最小分数。

数据范围

$N\le 40000$
$a_i\le 200$

思路

数据范围之大，DP显然不行。
我们使用 GarsiaWachs 算法（证明略）

假设第一堆左边有第零堆，第 $N$ 堆右边有第 $N + 1$ 堆，他们的石子数都是 $\infin$ ，便于计算。
每次从左往右遍历 $i$ ，找到第一个 $a_{i-1} < a_{i+1}$ 的位置
删除 $a_{i-1}、a_{i}$ ，记录分数 $K=a_{i-1}+a_i$
让 $j$ 从 $i - 2$ 向左遍历，找到第一个 $a_j>K$ 的位置
在 $j$ 之后插入 $K$ ，总得分增加 $K$
循环第 $2\sim5$ 步，直到找不到为止（即只剩下前后两堆 $\infin$ 和中间一堆）。
输出总得分即可。

核心代码

时间复杂度 $O(N\log N)$ ，可能会TLE，需要使用快读和O2优化等。

/*
 _            __   __          _          _
| |           \ \ / /         | |        (_)
| |__  _   _   \ V /__ _ _ __ | |     ___ _
| '_ \| | | |   \ // _` | '_ \| |    / _ \ |
| |_) | |_| |   | | (_| | | | | |___|  __/ |
|_.__/ \__, |   \_/\__,_|_| |_\_____/\___|_|
        __/ |
       |___/
*/
const int INF = 0x3f3f3f3f;

vector<int>V;
int work(){
    int id;
    int val;
    for(int i = 1;i < V.size() - 1;++i){
        if(V[i+1] > V[i-1]){
            val = V[i] + V[i-1];
            id = i-1;
            break;
        }
    }
    V.erase(V.begin() + id);
    V.erase(V.begin() + id);
    int id2;
    for(int i = id-1;i >= 0;--i){
        if(V[i] > val){
            id2 = i;
            break;
        }
    }
    V.insert(V.begin() + id2 + 1,val);
    return val;
}
int main()
{
    int n;n = read();
    V.push_back(INF);
    for(int i = 1;i <= n;++i){
        int t = read();
        V.push_back(t);
    }
    V.push_back(INF);
    ll ans = 0;
    for(int i = 1;i < n;++i){
        ans += work();
    }
    cout << ans;
    return 0;
}