1. 程式人生 > >CodeForces - 321E:Ciel and Gondolas (四邊形不等式優化DP)

CodeForces - 321E:Ciel and Gondolas (四邊形不等式優化DP)

題意:N個人排成一行,分成K組,要求每組的不和諧值之和最小。

思路:開始以為是斜率優化DP,但是每個區間的值其實已經知道了,即是沒有和下標有關的未知數了,所以沒必要用斜率。 四邊形優化。

dp[i][j]表示前j個人分為i組的最小代價。

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=4010;
int sum[maxn][maxn],cost[maxn][maxn],dp[810][maxn],pos[810][maxn];
void read(int &x){ x=0; char c=getchar(); while(c>'9'||c<'0') c=getchar(); while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar(); } int main() { int N,K,x,y; scanf("%d%d",&N,&K); rep(i,0,K) rep(j,0,N) dp[i][j]=2000000000; rep(i,1,N) rep(j,1,N){ read(sum[i][j]); sum[i][j]
=sum[i][j]+sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1]; } rep(i,1,N) rep(j,i+1,N) cost[i][j]=(sum[j][j]+sum[i-1][i-1]-sum[j][i-1]-sum[i-1][j])/2; rep(i,1,N) dp[1][i]=cost[1][i]; rep(i,2,K){ for(int j=N;j>=i;j--){ int L=pos[i-1][j]?pos[i-1][j]:1; int R=pos[i][j+1
]?pos[i][j+1]:N; rep(k,L,R){ if(dp[i-1][k]+cost[k+1][j]<dp[i][j]){ dp[i][j]=dp[i-1][k]+cost[k+1][j]; pos[i][j]=k; } } } } printf("%d\n",dp[K][N]); return 0; }