题意:给你一些人的屌丝值Di,计算总屌丝值=sum((k-1)*Di),k为第i个人进入的位置。给你一个栈,可以使人的位置改变。
方法一:记忆化搜索
dp[i][j][k]表示区间[i,j]第i个人放在k位置的屌丝值。
搜索所有的状态,记录已经搜索的状态。
代码:
//记忆化搜索
//这里也可以用二维的数组,另外用一个sum数组和。不过三维的实在是太神了
#include<iostream>
#include<vector>
#include<string>
#include<queue>
#include<map>
#include<cstdio>
#include<cstring>
#define maxn 105
#define INF 0xfffffff
#define min(a,b) a<b?a:b
#define max(a,b) a>b?a:b
using namespace std;
int n;
int dp[maxn][maxn][maxn];
int a[maxn];
int solve_dp(int s,int e,int k)
{
if(s>e) return 0;
if(s==e) return a[s]*(k-1);
if(dp[s][e][k]!=INF) return dp[s][e][k];//记忆化搜索
int i,j,thisk,nextk,tp,front,rear;
//将[s,e]划分为[s,i],[i+1,e],s可以放的位置从s-e
//前一个区间s放在最后,s+1放在原来k的位置
for(i=s;i<=e;i++)
{
thisk=k+(i-s);
nextk=thisk+1;
tp=a[s]*(thisk-1);
front=solve_dp(s+1,i,k);
rear=solve_dp(i+1,e,nextk);
dp[s][e][k]=min(dp[s][e][k],tp+front+rear);
}
return dp[s][e][k];
}
int main()
{
int t,i,j,k;
scanf("%d",&t);
int tt=1;
while(t--)
{
scanf("%d",&n);
for(i=1;i<=n;i++) scanf("%d",&a[i]);
for(i=0;i<=n;i++)
{
for(j=0;j<=n;j++)
{
for(k=0;k<=n;k++)
{
dp[i][j][k]=INF;//求最小值,所有的状态初始化为INF
}
}
}
printf("Case #%d: %d\n",tt++,solve_dp(1,n,1));
}
return 0;
}
直接用for循环,先把小区间计算出来,计算大区间时转化为小区间。
dp[i][j]表示区间[i,j]第i个人放在i位置的屌丝值。
代码:
//三for递推,把当前选择对后续选择的影响给提前计算了
//用sum数组来存和,能够在每次算区间的时候,把第一个看成1
#include<iostream>
#include<vector>
#include<string>
#include<queue>
#include<map>
#include<cstdio>
#include<cstring>
#define maxn 105
#define INF 0xfffffff
#define min(a,b) a<b?a:b
#define max(a,b) a>b?a:b
using namespace std;
int n;
int dp[maxn][maxn];
int a[maxn],sum[maxn];
int solve_dp()
{
int i,j,len,s,e,k;//dp[i][j]是把i作为第一个位置的值,思路也是把[i,j]区间分成两个
int tp,front,rear;
for(len=1;len<n;len++)//可以从小区间开始,大区间转为小区间
{
for(i=1;i+len<=n;i++)
{
s=i,e=i+len;
for(k=s;k<=e;k++)
{
tp=a[s]*(k-s);
front=dp[s+1][k]+dp[k+1][e];
rear=(sum[e]-sum[k])*(k-s+1);
dp[s][e]=min(dp[s][e],tp+front+rear);
}
}
}
return dp[1][n];
}
int main()
{
int t,i,j,k;
scanf("%d",&t);
int tt=1;
while(t--)
{
scanf("%d",&n);
memset(sum,0,sizeof(sum));
for(i=1; i<=n; i++)
{
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i];
}
for(i=1; i<=n; i++)
{
for(j=i+1; j<=n; j++)
{
dp[i][j]=INF;//求最小值,所有的状态初始化为INF
}
}
printf("Case #%d: %d\n",tt++,solve_dp());
}
return 0;
}