C. Gas Pipeline(递推 dp)

C. Gas Pipeline

题目链接

大致题意:

给出长度为n的01串,[0,n]的区间,建单位长度的管道代价a,建单位高度的柱子代价b,字符是0的

区间最低高度是1,字符是1的区间最低高度是2,求建造管道和柱子花费的最小值

---

解题思路:

分析:有四种建造情况,低管道代价是a+b,上升管道代价2*a+2*b,高管道代价a+2*b,下降管道代价2*a+b

(代价的计算是按照柱子前的管子+柱子,举例来说是[0,1]区间字符是0,代价是1位置的柱子加上[0,1]区间的管子)

状态表示:f[i][0]表示位置i时高度是1的最小代价,f[i][1]表示位置i时高度是2的最小代价

当字符是1,高度是2时,只能转移f[i][1],但i位置可以由i-1位置高度是1,或者是2转移得来
当字符是0,高度是1,两种状态都可以转移

转移方程:

字符是1:f[i][1] = min(f[i - 1][0] + 2 * a + 2 * b, f[i - 1][1] + a + 2 * b)

字符是0: f[i][0] = min(f[i - 1][0] + a + b, f[i - 1][1] + 2 * a + b)

f[i][1] = min(f[i - 1][0] + 2 * a + 2 * b, f[i - 1][1] + a + 2 * b)

---

AC代码:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
#define debug(a) cout << #a << " = " << a << endl;
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const int N = 1e6 + 10;
int n, m;
ll f[N][2]; //i位置高度为1的最小值  i位置高度为2的最小值
int main(void)
{
	ios::sync_with_stdio(0); cin.tie(0);
	int t; cin >> t;
	while (t--) {
		ll a, b; cin >> n >> a >> b; //a管道 b柱子
		char s[N]; cin >> s + 1;
		memset(f, 0x7f, sizeof f);
		f[0][0] = b; //最左边的一个柱子
		for (int i = 1; i <= n; ++i) {
			if (s[i] == '1' || s[i + 1] == '1')
				f[i][1] = min(f[i - 1][0] + 2 * a + 2 * b, f[i - 1][1] + a + 2 * b);
			else {
				f[i][0] = min(f[i - 1][0] + a + b, f[i - 1][1] + 2 * a + b);
				f[i][1] = min(f[i - 1][0] + 2 * a + 2 * b, f[i - 1][1] + a + 2 * b);
			}
		}
		cout << f[n][0] << endl;
	}
	return 0;
}

---