Codeforces Round #190 (Div. 1): E. Ciel and Gondolas（决策单调性DP+wqs二分）

 

E. Ciel and Gondolas

 

题意：

bzoj5311：https://www.lydsy.com/JudgeOnline/problem.php?id=5311

同一道题目，但是bzoj可能需要读入挂

 

思路：

决策单调性Ⅰ：四边形不等式

wqs二分

 

最无脑的暴力DP是O(n²k的)，dp[x][y]表示前x个贞鱼上了k辆车的怨气最小值，对于每个状态O(n)转移

→这个就不用多说了，入门DP，复杂度上天

 

有点经验的话可以看出（猜测）这个DP是有决策单调性的，这样的话就可以nlogn进行整体转移，复杂度O(nklogn)

很遗憾这样还是超时，因为k*n已经接近/超过亿级别，再加个logn肯定不行

 

想办法消掉k，这有个经典操作：wqs二分（https://www.cnblogs.com/CreeperLKF/p/9045491.html）

一般来讲用来应付一些这样的题目：给了一个选物品的限制条件，要求刚好选k个，让你最大化（最小化）权值（代价），然后其特点就是当选的物品越多的时候权值越大（越小）

一道非常经典的模板题如下：给你一个n个点m条边无向带权连通图，每条边是黑色或白色，让你求一棵最小权的恰好有k条白色边的生成树（bzoj2654）

正解就是二分权值val，让后让所有的白边权值全部加上val，跑一下Kruskal，然后看一下最小生成树中有多少条白色边

• 如果白色边个数>k，增加val的值继续二分
• 如果白色边个数<k，减少val的值继续二分

直到最小生成树中刚好有k条白边位置，直接减去val*k就是答案

这道题当然如出一辙：二分val，每多选一辆车就加上额外val点代价，看最后DP完之后选了多少辆车，如果不到k辆就减少val，否则增大val，搞定，时间复杂读O(nlognlogW)，其中W为val最大可能

 

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<map>
#include<string>
#include<math.h>
#include<queue>
#include<stack>
#include<iostream>
using namespace std;
#define LL long long
#define mod 1000000007
int n, k, sum[4005][4005];
typedef struct Res
{
	int id;
	int l, r;
}Res;
Res q[4005], now;
int dp[4005], cnt[4005];
int Cal(int x, int y)
{
	return (sum[y][y]-sum[x-1][y]-sum[y][x-1]+sum[x-1][x-1])/2;
}
int Check(int i, int j, int k)
{
	int a, b;
	a = dp[i]+Cal(i+1, k);
	b = dp[j]+Cal(j+1, k);
	if(a<b || a==b && cnt[i]<cnt[j])
		return 1;
	return 0;
}
int Sech(int i, int j, int c)
{
	int l, r, m;
	l = q[c].l, r = n;
	while(l<r)
	{
		m = (l+r)/2;
		if(Check(i, j, m))
			r = m;
		else
			l = m+1;
	}
	return r;
}
int Jud(int val)
{
	int L, R, i, x;
	L = 1, R = 0;
	now.id = 0, now.l = 1, now.r = n;
	q[++R] = now;
	for(i=1;i<=n;i++)
	{
		if(q[L].l>q[L].r)
			L++;
		q[L].l++;
		dp[i] = dp[q[L].id]+Cal(q[L].id+1, i)+val;
		cnt[i] = cnt[q[L].id]+1;
		if(R>=L && Check(q[R].id, i, n))
			continue;
		while(R>=L && Check(i, q[R].id, q[R].l))
			R--;
		if(R>=L)
		{
			x = Sech(i, q[R].id, R);
			q[R].r = x-1;
			now.id = i, now.l = x, now.r = n;
		}
		else
			now.id = i, now.l = i+1, now.r = n;
		q[++R] = now;
	}
	return cnt[n];
}
int main(void)
{
	int i, j, x, l, r, m;
	scanf("%d%d", &n, &k);
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=n;j++)
		{
			scanf("%d", &x);
			sum[i][j] = sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1]+x;
		}
	}
	l = 0, r = sum[n][n];
	while(l<r)
	{
		m = (l+r)/2;
		if(Jud(m)<=k)
			r = m;
		else
			l = m+1;
	}
	Jud(r);
	printf("%d\n", dp[n]-k*r);
	return 0;
}
/*
9 2
0 0 0 0 1 1 1 1 1
0 0 0 0 1 1 1 1 1
0 0 0 0 1 1 1 1 1
0 0 0 0 1 1 1 1 1
1 1 1 1 0 0 1 1 1
1 1 1 1 0 0 1 1 1
1 1 1 1 1 1 0 0 0
1 1 1 1 1 1 0 0 0
1 1 1 1 1 1 0 0 0
*/