【URAL 1900】Brainwashing Device(预处理区间和+DP)
题目大意:
n个城市1~n,第i座城市与i+1有一条路。
第i个城市往i+1~n每座城市都有客流量,每个人如果从i->j,要经过i与j间每一条边一次。
政府决定缩短k条道路的距离(1 <= k <= n-1)
对于每个人,从i到j,如果中间任何一条道路被缩短,他会be happy。
问最多让多少人be happy,然后输出缩短的道路编号,编号i表示i->i+1这条路
考虑dp[i][j]表示缩短边i,同时总缩短了j条边的最大happy值。
再一个存它的前驱。
这样最后输出递归输出即可。
典型的路径记忆dp,ural后面这类题还真是多……
不过有点恶心的是转移。dp[j][l-1]->dp[i][l],增加的happy值是j+1~i这几座城市,通往i+1及后面的城市的客流量。。。
我做法是dp枚举j的过程中累积,这样只需要记录每个点i,通往点j及以后城市的客流量,输入完倒着走一边就可以了。
代码如下:
#include <iostream>
#include <cmath>
#include <vector>
#include <cstdlib>
#include <cstdio>
#include <climits>
#include <ctime>
#include <cstring>
#include <queue>
#include <stack>
#include <list>
#include <algorithm>
#include <map>
#include <set>
#define LL long long
#define Pr pair<int,int>
#define fread(ch) freopen(ch,"r",stdin)
#define fwrite(ch) freopen(ch,"w",stdout)
using namespace std;
const int INF = 0x3f3f3f3f;
const int mod = 1e9+7;
const double eps = 1e-8;
const int maxn = 555;
int dp[2][maxn][maxn];
int tr[maxn][maxn];
void prt(int u,int k)
{
if(k == 0) return;
prt(dp[0][u][k],k-1);
printf("%d ",u);
}
int main()
{
fread("in.in");
//fwrite("");
int n,k,x;
scanf("%d%d",&n,&k);
for(int i = 1; i <= n; ++i)
{
for(int j = i+1; j <= n; ++j)
scanf("%d",&tr[i][j]);
for(int j = n-1; j > i; --j)
tr[i][j] += tr[i][j+1];
}
memset(dp,-1,sizeof(dp));
dp[0][0][0] = 0;
dp[1][0][0] = 0;
for(int i = 1; i < n; ++i)
{
int tmp = tr[i][i+1];
dp[0][i][0] = dp[1][i][0] = 0;
for(int l = i-1; l >= 0; --l)
{
for(int j = 1; j <= k; ++j)
{
if(dp[0][l][j-1] == -1) break;
if(dp[1][l][j-1]+tmp > dp[1][i][j])
{
dp[0][i][j] = l;
dp[1][i][j] = dp[1][l][j-1]+tmp;
}
}
tmp += tr[l][i+1];
}
}
int id = 1;
for(int i = 1; i < n; ++i)
{
if(dp[1][i][k] >= dp[1][id][k]) id = i;
}
printf("%d\n",dp[1][id][k]);
prt(id,k);
return 0;
}