【SPOJ-ALIEN2】Aliens at the train, again!【DP】

题意:

有两个火车站A和B,每个有n个站台,每个站台有一些人。外星人从1号站台开始出发(他可以选择从A或者B开始),每次他可以继续沿着当前火车站走,或者切换到另一个火车站。如果在第i号站台切换,那么遇到的人数就是Ai+Bi。但是外星人遇到的人数总和不能超过k。问你外星人最多能走到几号站台,最少能遇到多少人。(如果下一步遇到人数总和大于k,那么外星人会在当前站台下车。)


比较简单的dp。


#include <cstdio>
#include <algorithm>

using namespace std;

const int maxn = 10005, inf = 0x3f3f3f3f;

int n, k, a[maxn], b[maxn], dp[maxn][2];

inline int iread() {
	int f = 1, x = 0; char ch = getchar();
	for(; ch < '0' || ch > '9'; ch = getchar()) f = ch == '-' ? -1 : 1;
	for(; ch >= '0' && ch <= '9'; ch = getchar()) x = x * 10 + ch - '0';
	return f * x;
}

int main() {
	n = iread(); k = iread();
	for(int i = 1; i <= n; i++) a[i] = iread();
	for(int i = 1; i <= n; i++) b[i] = iread();

	dp[1][0] = a[1]; dp[1][1] = b[1];
	for(int i = 2; i <= n; i++) {
		dp[i][0] = dp[i - 1][0] + a[i]; dp[i][1] = dp[i - 1][1] + b[i];
		dp[i][0] = min(dp[i][0], dp[i][1] + a[i]);
		dp[i][1] = min(dp[i][1], dp[i][0] + b[i]);
		if(dp[i][0] > k && dp[i][1] > k) {
			printf("%d %d\n", i - 1, min(dp[i][0] - a[i], dp[i][1] - b[i]));
			return 0;
		}
	}

	printf("%d %d\n", n, min(dp[n][0], dp[n][1]));
	return 0;
}


你可能感兴趣的:(dp)