BZOJ 1560 火星藏宝图（DP）

struct node
{
    int x,y,w;
};


node a[N];
int b[1005],c[1005],n,m;


int cmp(node a,node b)
{
    if(a.x!=b.x) return a.x<b.x;
    return a.y<b.y;
}


int dis(int x1,int y1,int x2,int y2)
{
    return sqr(x1-x2)+sqr(y1-y2);
}


int main()
{
    RD(n,m);
    int i,j;
    FOR1(i,n) RD(a[i].x,a[i].y,a[i].w);
    sort(a+1,a+n+1,cmp);
    int x,y,w,temp;
    FOR1(i,m) b[i]=-INF;
    b[1]=a[1].w; c[1]=1;
    for(i=2;i<=n;i++)
    {
        x=a[i].x; y=a[i].y; w=a[i].w;
        temp=-INF;
        for(j=1;j<=y;j++) upMax(temp,b[j]-dis(x,y,c[j],j));
        temp+=w;
        b[y]=temp; c[y]=x;
    }
    PR(temp);
}