【线性dp】邮局选址+bus

UPC2020年春混合个人训练 5.1日场

邮局选址
时间限制: 1 Sec 内存限制: 128 MB
题目描述
有n个村庄分布在一条直线上，每个村庄可以用一个坐标xi来进行描述。现在，你需要建设m个邮局，使得每个村庄到离它最近的邮局的距离之和最小。
输入
第一行两个正整数n,m。第二行n个递增的整数，表示x1~xn。
输出
输出一行一个整数，为最小的距离之和。
样例输入
10 5
1 2 3 6 7 9 11 22 44 50
样例输出
9
提示
对于100%的数据，满足1≤n≤300，1≤m≤30，1≤xi≤10000。
解题思路：如果这个题是建一个邮局的话，那比较好说，如果村庄数是奇数时，邮局的位置就是中位数的位置，偶数时，邮局的位置取中间的两个数的位置都可；现在是建m个邮局，不妨用dp，dp[i][j]表示前i个村庄建立j个邮局的最优解，举个一般情况来说，dp[i][j]是由dp[k][j-1]+d[k+1][i]得到的，这里解释一下，d数组是预处理出任何两个村庄之间建立一个邮局的最优解，k的范围是（1，i）,当你建立第j个邮局时，第j个邮局的贡献区间肯定是和前面组成一个部分，最小是他自己，dp[k][j-1]在此之前就求得了，动态转移方程就是 dp[i][j]=min(dp[i][j],dp[k][j-1]+d[k+1][i]) 遍历一遍k取min就好了，最后的ans就是dp[n][m]啦
AC源：

#include<bits/stdc++.h>
#include<vector>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
typedef long long ll;
const int maxn = 1e6+7;
typedef pair<int,int> PII;
PII q[100000];
int n,m;
int a[maxn],d[1000][1000],dp[1000][1000];
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)   ///预处理出任何村庄之前的最优解
    {
        for(int j=i+1;j<=n;j++)
        {
            int flag=a[(i+j)/2];
            for(int k=i;k<=j;k++)
            {
                d[i][j]+=abs(a[k]-flag);
            }
        }
    }
    for(int i=1;i<=n;i++)   ///先求一波前i个村庄建一个邮局的情况
    {
        dp[i][1]=d[1][i];
    }
    for(int i=2;i<=n;i++)
    {
        for(int j=2;j<=min(i,m);j++)  ///这里j的范围要注意一下
        {
            dp[i][j]=maxn;   ///先设为一个比较大的数
            for(int k=1;k<i;k++)
                dp[i][j]=min(dp[i][j],dp[k][j-1]+d[k+1][i]);   ///动态转移方程
        }
    }
    cout << dp[n][m];
    return 0;
}

bus
时间限制: 1 Sec 内存限制: 128 MB
题目描述
两个球队的支持者要一起坐车去看球，他们已经排成了一列。我们要让他们分乘若干辆巴士，同一辆巴士上的人必须在队伍中是连续的。为了在车上不起冲突，希望两队的支持者人数尽量相等，差至多是D。有一个例外，就是一辆车上的人全部都是一个球队的支持者。问要将这N个人全部送至球场，至少要几辆巴士。
输入
第一行是整数N和D。
接下来一行，按排队的顺序，描述每个人支持的球队，用H或J表示。该行没有任何多余的字符。
输出
一个整数，表示要多少巴士。
样例输入
14 3
HJHHHJHJHHHHHH
样例输出
2
提示
对于100%的数据：N,D≤2500，数据有合理的梯度。
解题思路：我感觉这个题和上个题得思路是相同的，好像还简单点，一维就够用，dp[i]表示前i个人可用的最少的巴士的数量，加入一个人时，他有可能直接与前面的所有人组成合理解，这种情况也是当下的最优解，也就是一辆巴士；也有可能和前面的一段组成合理解，所以这里也要预处理一下，开一个数组表示出任意两个人之间可不可以组成合理解，如果是合理解，赋值为1，否则为0，题目也说是连续的了，所以动态转移方程也就出来了，见下面的代码→
AC源：

动态转移方程：
		if(d[1][i]==1) dp[i]=1;   ///d数组是预处理的数组
        else{
            dp[i]=maxn;   ///先给个比较大的数
            for(int j=2;j<=i;j++)
            {
                if(d[j][i]==1) dp[i]=min(dp[i],dp[j-1]+1);   ///j--i能组成合理解，dp[j-1]也早已求得，遍历取min就好了
            }
        }
#include<bits/stdc++.h>
#include<vector>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
typedef long long ll;
const int maxn = 1e6+7;
typedef pair<int,int> PII;
PII q[100000];
int n,m;
char s[maxn];
int dp[maxn],d[3000][3000];
int main(){
    scanf("%d%d",&n,&m);
    scanf("%s",s+1);
    memset(d,0,sizeof d);
    for(int i=1;i<=3000;i++) d[i][i]=1;   ///一定是合理解
    for(int i=1;i<=n;i++)
    {
        int ans1=0,ans2=0;
        if(s[i]=='H') ans1++;
        else ans2++;
        for(int j=i+1;j<=n;j++)
        {
            if(s[j]=='H') ans1++;
            else ans2++;
            if(abs(ans1-ans2)<=m||ans1==0||ans2==0) d[i][j]=1;  ///巴士车上如果全是一队的人也是合理解
            ///else break;
        }
    }
    dp[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(d[1][i]==1) dp[i]=1;
        else{
            dp[i]=maxn;
            for(int j=2;j<=i;j++)
            {
                if(d[j][i]==1) dp[i]=min(dp[i],dp[j-1]+1);
            }
        }
        ///cout << dp[i] << endl;
    }
    cout << dp[n];
    return 0;
}

做了这两个线性dp的题还是大有体会的
成长这一路就是懂得闭嘴努力，知道低调谦逊，学会强大自己，在每一个值得珍惜的日子里，拼命去成为自己想成为的人。加油！！！