uva 116(dp)

题意:一个二维数组内有一些值(有正数有负数),求出任意第一列开始到任意最后一列结束经过的路径和最小,打印这个最小和和这条路径的行数。

题解:dp,用f的二维数组保存每一步到最后一列的最小值,从第一列到最后一列递归计算出f数组(注意不能从最后一列开始递归,因为要字典序最小),递归结束得到f数组后,比较第一列的f数组的最小值,最后根据这个数字减去这个位置上数组的数找到下一个位置,将每一个位置的行数保存后输出。

因为数组内的数字可能为负,所以f初始化时需要注意放一个极大值或极小值,不影响计算。


#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = 105;
const long long MAX = 922337203685477580;
const long long MIN = -922337203685477580;
int row, col, res[N];
long long f[N][N], arr[N][N];
int flag[3] = {-1, 0, 1};

long long dp(int r, int st) {
	if (f[r][st] != MIN) 
		return f[r][st];
	f[r][st] = arr[r][st];
	if (st == col - 1)
		return arr[r][st];
	long long minn = MAX, temp;
	for (int i = 0; i < 3; i++) {
		if (r + flag[i] == -1)
			temp = dp(row - 1, st + 1);
		else if (r + flag[i] == row)
			temp = dp(0, st + 1);
		else
			temp = dp(r + flag[i], st + 1);
		if (minn > temp)
			minn = temp;
	}
	f[r][st] += minn;
	return f[r][st];
}

void print_path(int r, int c) {
	long long n = f[r][c] - arr[r][c];
	int temp[3];
	res[c] = r + 1;
	if (c == col - 1) {
		printf("%d", res[0]);
		for (int i = 1; i < col; i++)
			printf(" %d", res[i]);
		printf("\n");
		return;
	}
	for (int i = 0; i < 3; i++) {
		temp[i] = r + flag[i];
		if (temp[i] == -1)
			temp[i] = row - 1;
		else if (temp[i] == row)
			temp[i] = 0;
	}
	sort(temp, temp + 3);
	for (int i = 0; i < 3; i++)
		if (f[temp[i]][c + 1] == n) {
			print_path(temp[i], c + 1);
			break;
		}
}

int main() {
	while (scanf("%d%d", &row, &col) != EOF) {
		for (int i = 0; i < N; i++)
			for (int j = 0; j < N; j++)
				f[i][j] = MIN;
		for (int i = 0; i < row; i++)
			for (int j = 0; j < col; j++)
				scanf("%lld", &arr[i][j]);
		for (int i = 0; i < row; i++)
			dp(i, 0);
		long long minn = f[0][0];
		int k = 0;
		for (int i = 1; i < row; i++) {
			if (minn > f[i][0]) {
				minn = f[i][0];
				k = i;
			}
		}
		print_path(k, 0);
		printf("%lld\n", minn);
	}
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值