5843：摘樱桃（N个数分组，转化超级原点，SPFA最短路）

5843: 摘樱桃

时间限制: 1 Sec   内存限制: 128 MB
提交: 98   解决: 46
[ 提交][ 状态][ 讨论版][命题人: admin]

题目描述

有n个樱桃排成一列，第i个樱桃的甜度为v[i]，你要把n个樱桃分成若干组，其中每一组的樱桃必须相邻。每一组樱桃的美味度为(sum-T)^2，其中sum是这组樱桃的甜度之和，T为输入给定的系数。
一组方案的美味度为每一组的美味度之和。
求可行方案最小的美味度。

输入

第一行两个正整数n，T。
第二行n个整数，第i个整数是第i个樱桃的甜度，v[i]。

输出

一行一个非负整数，为最小美味度。

样例输入

5 5
3 5 2 1 6

样例输出

9

提示

对于50%的数据满足 n<=10,T<=1000,v[i]<=10
对于70%的数据满足 n<=100
对于所有数据，n<=10^3,T<=1000,v[i]<=10

来源

2018山东冬令营 

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
#include<vector>
#define inf 0x3f3f3f3f 
using namespace std;
int n, t, book[1005], sum[1005], head[1005];
long long dis[1005];
struct node{
    long long to, w, next;
}edge[1000005];
void spfa()
{
    memset(book, 0, sizeof(book));
    memset(dis, inf, sizeof(dis));
    queue<int>q;
    book[0]=1;
    dis[0]=0;
    q.push(0);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        book[x]=0;
        for(int i=head[x];i!=-1;i=edge[i].next)
        {
            long long to=edge[i].to, w=edge[i].w;
            if(dis[to]>dis[x]+w)
            {
                dis[to]=dis[x]+w;
                if(!book[to])
                {
                    book[to]=1;
                    q.push(to);
                } 
            }
        }
    }
}
int main()
{
    sum[0]=0;int cnt=0;
    memset(head, -1, sizeof(head));
    scanf("%d%d", &n, &t);
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d", &x);
        sum[i]=sum[i-1]+x;
    }
    for(int i=0;i<=n;i++)
        for(int j=i+1;j<=n;j++)
        {
            long long w=(sum[j]-sum[i]-t)*(sum[j]-sum[i]-t);
            edge[cnt].to=j;
            edge[cnt].w=w;
            edge[cnt].next=head[i];
            head[i]=cnt++;
        }
    spfa();
    printf("%lld\n", dis[n]);
    return 0;
} 
/**************************************************************
    Problem: 5843
    User: ldu_reserver201701
    Language: C++
    Result: 正确
    Time:52 ms
    Memory:24712 kb
****************************************************************/