[UOJ240]aliens

学习了一下凸优化DP,感觉挺有意思的

首先把所有点对称到左下角,然后以每个点为顶点画等腰直角三角形,将被覆盖的点去掉,现在所有点从左上到右下横纵坐标都是递增的,设坐标为$(x_{1\cdots M},y_{1\cdots M})$

设$f_{j,i}$表示拍$j$次照片覆盖$i$个点的最少覆盖方格数,枚举最后一个矩形覆盖到之前的哪个点,有$f_{j,i}=\min\limits_{0\leq k\lt i}f_{j-1,k}+\left(y_i-x_{k+1}+1\right)^2-\max\left(y_i-x_{i+1}+1,0\right)^2$,最后减去的是和下一个矩形的重复部分

直接斜率优化可以$O(nk)$,还需要再快一些

凸优化用于解决一类“恰好选$k$个”的问题,在这题中,如果把$f_{j,i}$关于$j$的图像画出来,可以看出它的斜率不减(这我不会证,官方题解上也没讲怎么证,但一般可以通过打差分表猜出结论)

对于常数$C$,设$g_i=\min\limits_jf_{j,i}+jC$,将$f$用转移展开后再回代$g$的定义,我们得到$g_i=\min\limits_{0\leq k\lt i}g_k+\left(y_i-x_{k+1}+1\right)^2-\max\left(y_i-x_{i+1}+1,0\right)^2+C$,可以$O(n)$求

$g_i$是将$\Delta f_{j,i}$平移后取$f$的最小值得到的,所以当$C$增大时使$g$取到最小值的$j$会变小,于是我们可以通过二分找到一个$p$使得当$C=p$时,使$g_M$取得最小值的$j_1\geq k$,当$C=p+1$时,使$g_M$取得最小值的$j_2\leq k$

我们已经得到了$f_{j_1,M},f_{j_2,M}$,现在要求$f_{k,M}\left(k\in[j_2,j_1]\right)$,因为再把$\Delta f_{j,i}$往上平移$1$单位就会让最小值位置偏移,所以$\Delta f_{j_2+1\cdots j_1,M}$全部相等,也就是说$f_{j_2\cdots j_1,M}$是一条直线,所以可以直接算出$f_{k,M}$

总的来说,如果$f_{j,i}$是关于$j$的凸函数,难以求值却易求最值,那么可以考虑偏移$\Delta f_{j,i}$来改变最小值的位置,进而求出某个位置的值

总时间复杂度$O(n\log m)$,相当优美

#include<stdio.h>
#include<vector>
#include<math.h>
using namespace std;
typedef long long ll;
typedef double du;
typedef vector<int> vi;
const du eps=1e-5;
template<class t>void gmax(t&a,t b){
	if(a<b)a=b;
}
ll sqr(ll x){return x*x;}
struct line{
	ll k,b;
	int i;
	line(ll k=0,ll b=0,int i=0):k(k),b(b),i(i){}
	ll v(ll x){return k*x+b;}
}q[100010],u;
du its(line a,line b){
	return(b.b-a.b)/(du)(a.k-b.k);
}
int x[100010],y[100010],M;
ll g[100010];
int b[100010];
int get(ll c){
	int head,tail,i;
	head=tail=1;
	q[1]=line((x[1]-1)*-2,sqr(x[1]-1));
	for(i=1;i<=M;i++){
		while(head<tail&&its(q[head],q[head+1])<y[i]-eps)head++;
		if(head<tail&&fabs(its(q[head],q[head+1])-y[i])<eps){
			b[i]=min(q[head].i,q[head+1].i)+1;
			head++;
		}else
			b[i]=q[head].i+1;
		g[i]=q[head].v(y[i])+sqr(y[i])+c-sqr(max(y[i]-x[i+1]+1,0));
		if(i<M){
			u=line((x[i+1]-1)*-2,g[i]+sqr(x[i+1]-1),b[i]);
			while(head<tail&&its(q[tail],q[tail-1])>its(q[tail],u))tail--;
			q[++tail]=u;
		}
	}
	return b[M];
}
int a[1000010];
ll inter(ll lc,ll rc,ll x){
	ll lx,ly,rx,ry;
	lx=get(lc);
	ly=g[M]-lc*lx;
	rx=get(rc);
	ry=g[M]-rc*rx;
	if(lx!=rx)
		return(ly-ry)/(rx-lx)*(rx-x)+ry;
	else
		return ly;
}
ll take_photos(int n,int m,int k,vi r,vi c){
	int i,t;
	ll l,_r,mid,ans;
	for(i=0;i<n;i++){
		r[i]++;
		c[i]++;
		if(r[i]<c[i])
			gmax(a[r[i]],c[i]);
		else
			gmax(a[c[i]],r[i]);
	}
	M=0;
	t=0;
	for(i=1;i<=m;i++){
		if(a[i]>t){
			M++;
			x[M]=i;
			y[M]=a[i];
			t=a[i];
		}
	}
	x[M+1]=m+1;
	k=min(k,M);
	#define r _r
	l=0;
	r=(ll)m*m;
	ans=0;
	while(l<=r){
		mid=(l+r)>>1;
		if(get(mid)>=k){
			ans=mid;
			l=mid+1;
		}else
			r=mid-1;
	}
	return inter(ans+1,ans,k);
	#undef r
}

新的一年里要继续努力啊...

posted @ 2019-01-04 10:57 jefflyy 阅读(...) 评论(...) 编辑 收藏