HDU3480

题意:给你n个数,然后让你分成m个集合,每个集合有一个值(最大值减最小值,然后平方),求整个集合的可能最小值。

思路:因为每个集合里的值只和最大和最小值有关,所以很容易想到先排序,然后用DP可求得解,状态转移方程dp[i][j] = min(dp[i][j] , dp[k][j - 1] + (a[i] - a[k + 1]) ^ 2),j表示数组的下标,i表示集合数,dp为最小值,但是因为n为10000,m为5000,这个复杂度肯定会超时,所以可以用斜率或四边形优化来减小复杂度,式子就不证明了,这里直接引用,另外贴上四边形优化的一个链接,讲的挺详细的点击打开链接

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 10010
#define ll long long
using namespace std; ll a[MAXN];
ll dp[MAXN][MAXN];
int N, M;
ll q[MAXN];
ll head, tail; void init()
{
for (int i = 1; i <= N; i++)
dp[1][i] = (a[i] - a[1])*(a[i] - a[1]);
} void solve()
{
for(int i=2;i<= M;i++)
{
head=tail=0;
q[tail++]=i-1;
for(int j = i;j <= N;j++)
{
while(head+1<tail)
{
int p1=q[head];
int p2=q[head + 1];
int x1=a[p1 + 1];
int x2=a[p2 + 1];
int y1=dp[i - 1][p1] + x1*x1;
int y2=dp[i - 1][p2] + x2*x2;
if((y2-y1)<= 2*a[j]*(x2-x1))head++;
else break;
}
int k=q[head];
dp[i][j]=dp[i-1][k]+(a[j]-a[k+1])*(a[j]-a[k+1]);
while(head+1<tail)
{
int p1=q[tail-2];
int p2=q[tail-1];
int p3=j;
int x1=a[p1+1];
int x2=a[p2+1];
int x3=a[p3+1];
int y1=dp[i-1][p1]+x1*x1;
int y2=dp[i-1][p2]+x2*x2;
int y3=dp[i-1][j]+x3*x3;
if ((y3-y2)*(x2-x1)<=(y2-y1)*(x3-x2))tail--;
else break;
}
q[tail++]=j;
}
}
}
int main()
{
int T;
scanf("%d",&T);
int k=0;
while(T--)
{
k++;
scanf("%d%d",&N,&M);
for (int i = 1; i <= N; i++)
scanf("%d", &a[i]);
sort(a+1,a+N+1);
init();
solve();
printf("Case %d: %d\n",k,dp[M][N]);
}
return 0;
}
上一篇:【一天一道LeetCode】#76. Minimum Window Substring


下一篇:【解决办法】糟糕,我的电脑只有IE64位浏览器能上网,其他软件都上不了网