算法提高 宰羊

题目描述

        炫炫回了内蒙，肯定要吃羊肉啦，所有他家要宰羊吃。

        炫炫家有N只羊，羊圈排成一排，标号1~N。炫炫每天吃掉一只羊（这食量！其实是放生啦），吃掉的羊的邻居会以为它被放生了，然后又会告诉他们的邻居，这样一直传播下去，除非某个邻居已经被“放生”了。每一天，所有知道某羊被“放生”了这个消息的羊都会很不满，如果不给他们巧克力的话，他们就会很造反，炫炫已经知道他要吃掉哪些羊，他可以任意安排吃的顺序，然后使巧克力的用量最小，请求出这个最小值。

输入格式
        本题有多组数据，第一行为数据组数T。
        对于每组数据
        第一行：两个用空格隔开的整数：N和M，表示羊的数量和需要吃掉的数量
        第二行：有M个数，表示要吃那些羊。

输出格式
        T行，为每组数据的答案。

样例输入

2
8 1
3
20 3
3 6 14

样例输出
7
35

数据规模和约定
        T=10
        N<=10000
        M<=100

---

分析:

        当选择一只羊的时候区间被分成两个区间，形成同样的形式的两个子问题, 且选择这只羊所产生的巧克力数等于区间内样的个数减1.

• 状态:

$dp(l,r)$ 表示$[l,r]$范围的最优解

• 状态转移方程:

        假设从$[l,r]$ 区间内选择两只羊分别为 $i1,i2$(均在区间范围内), 则状态转移方程为

$dp(l,r)=min[dp(l,i1-1)+dp(i1+1,r),dp(l,i2-1)+dp(i2+1,r)]+(r-l)$

---

代码:

#include <bits/stdc++.h>
using namespace std;
int M[100];
int n, m;

int dfs(map<pair<int, int>, int> &mp,int l, int r)
{
    if(r <= l) return 0;
    pair<int, int> p = make_pair(l, r);
    if(mp.find(p) != mp.end()) return mp[p];          // 当前状态访问过了
    int ans = INT_MAX;                                // 获取当前最优答案
    for(int i = 0; i < m && M[i] <= r; ++i) 
    {
        if(M[i] < l) continue;    // make sure : l <= M[i] <= r
        ans = min(ans, dfs(mp, l, M[i] - 1) + dfs(mp , M[i] + 1, r));
    }
    if(ans == INT_MAX) ans = 0;
    else ans = ans + (r - l);
    return mp[p] = ans;
} 
int main()
{
#ifdef LOCAL
    freopen("ADV-299.in", "r", stdin);
#endif
    std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0); 
    int T;
    cin >> T;
    while(T--)
    {
        cin >> n >> m;
        for(int i = 0; i < m; ++i) cin >> M[i];
        map<pair<int, int>, int> mp;
        cout << dfs(mp ,1, n) << endl;
    }
    return 0;
}