1. 程式人生 > >洛谷4056 [JSOI2009]火星藏寶圖(斜率優化+dp)

洛谷4056 [JSOI2009]火星藏寶圖(斜率優化+dp)

qwq又要吐槽一句我菜的真實。

由於網上很多 O ( n m ) O(nm) 做法,這裡就不做贅述了。

我這裡只寫一下 O

( m 2 ) O(m^2) 的做法。

首先我們定義一個 d p [

i ] [ j ] dp[i][j] 表示到 ( i ,
j ) (i,j)
這個座標的島的最大收益。

然後我們考慮轉移,首先對於同一列來說,只有截止到 i i 行之前的行數最大的那個點才有可能作為轉移點。

在這裡插入圖片描述

這樣一個圖,如果我們從上面的點轉移的代價是 ( a + b ) 2 c 2 + v a l 2 -(a+b)^2-c^2+val_2

如果從下面那個點呢,則是 a 2 b 2 c 2 + v a l 2 + v a l 1 -a^2-b^2-c^2+val_2+val_1

顯然從下面的點轉移,代價更小,而且還能獲得兩個點的收益。

那麼我們不妨定義 p o s [ i ] pos[i] 表示 i i 這一列最底下的行是多少。

那麼 d p [ i ] [ p ] = m a x ( f [ i ] [ j ] ( j p ) 2 ( i p o s [ j ] ) 2 + w [ i ] [ p ] ) ) dp[i][p]=max(f[i][j]-(j-p)^2-(i-pos[j])^2+w[i][p]))
經過一番推柿子,我們設 g [ j ] = f [ i ] [ j ] ( i p o s [ j ] ) 2 j 2 g[j]=f[i][j]-(i-pos[j])^2-j^2

g ( j ) g ( k ) j k > 2 p \frac{g(j)-g(k)}{j-k} > -2p

直接斜率優化
然後列舉行,列舉列,分別進行轉移就行

有一個小細節就是對於一個位置 ( i , j ) (i,j) ,如果他有島,那麼 d p [ p o s [ j ] ] [ j ] d p [ i ] [ j ] dp[pos[j]][j]和dp[i][j] 都要加入佇列
qwq

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#define mk make_pair
#define ll long long
#define int long long
using namespace std;
inline int read()
{
  int x=0,f=1;char ch=getchar();
  while (!isdigit(ch)) {if (ch=='-') f=-1;ch=getchar();}
  while (isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
  return x*f;
}
const int maxn = 2010;
struct Point{
	int x,y,hang,line;
};
int dp[maxn][maxn];
int pos[maxn]; // pos[i]表示i這一列最靠下的點的行數是多少.
int val[maxn][maxn];
int n,m;
Point q[maxn*10];
int chacheng(Point x,Point y)
{
	return x.x*y.y-x.y*y.x;
	
}
bool count(Point i,Point j,Point k)
{
	Point x,y;
	x.x=k.x-i.x;
	x.y=k.y-i.y;
	y.x=k.x-j.x;
	y.y=k.y-j.y;
	if (chacheng(x,y)>=0) return true;
	return false;
}
int head=1,tail=0;
void push(Point x)
{
	while (tail>=head+1 && count(q[tail-1],q[tail],x))	tail--;
	q[++tail]=x;
}
void pop(int lim)
{
	while (tail>=head+1 && q[head+1].y-q[head].y>lim*(q[head+1].x-q[head].x)) head++;
}
signed main()
{
  n=read(),m=read();
  for (int i=1;i<=n;i++)
  {
  	 int x=read(),y=read();
  	 val[x][y]=read();
  }
  dp[1][1]=val[1][1];
  pos[1]=1;
  push((Point){1,dp[1][1]-1,1,1});
  for (int i=1;i<=m;i++)
  {
  	 head=1,tail=0;
  	 for (int j=1;j<=m;j++)
  	 {
  	   if (i==1 &&j==1) continue;
  	   if (pos[j])
  	   	 push((Point){j,dp[pos[j]][j]-(pos[j]-i)*(pos[j]-i)-j*j,pos[j],j});
  	   if (val[i][j])
  	   {
  	   	  pop((-2)*j);
  	   	  Point now = q[head];
  	   	  dp[i][j]=dp[now.hang][now.line]-(now.line-j)*(now.line-j)-(now.hang-i)*(now.hang-i)+val[i][j];
  	   	  pos[j]=i;
		  push((Point){j,dp[pos[j]][j]-(pos[j]-i)*(pos[j]-i)-j*j,pos[j],j});
		 }
	 }
  }
  cout<<dp[m][m]<<endl; 
  return 0;
}