【APIO2015】Palembang Bridges

题目描述

一条东西走向的穆西河将巴邻旁市一分为二,分割成了区域 $A$ 和区域 $B$。

每一块区域沿着河岸都建了恰好 $1000000001$ 栋的建筑,每条岸边的建筑都从 $0$ 编号到 $1000000000$。相邻的每对建筑相隔 $1$ 个单位距离,河的宽度也是 $1$ 个单位长度。区域 $A$ 中的 $i$ 号建筑物恰好与区域 $B$ 中的 $i$ 号建筑物隔河相对。

城市中有 $N$ 个居民。第 $i$ 个居民的房子在区域 $P_i$ 的 $S_i$ 号建筑上,同时他的办公室坐落在 $Q_i$ 区域的 $T_i$ 号建筑上。一个居民的房子和办公室可能分布在河的两岸,这样他就必须要搭乘船只才能从家中去往办公室,这种情况让很多人都觉得不方便。为了使居民们可以开车去工作,政府决定建造不超过 $K$ 座横跨河流的大桥。

由于技术上的原因,每一座桥必须刚好连接河的两岸,桥梁必须严格垂直于河流,并且桥与桥之间不能相交。

当政府建造最多 $K$ 座桥之后,设 $D_i$ 表示第 $i$ 个居民此时开车从家里到办公室的最短距离。请帮助政府建造桥梁,使得 $D_1 + D_2 + \dots + D_N$ 最小。

输入格式

输入的第一行包含两个正整数 $K$ 和 $N$,分别表示桥的上限数量和居民的数量。

接下来 $N$ 行,每一行包含四个参数:$P_i, S_i, Q_i$ 和 $T_i$,表示第 $i$ 个居民的房子在区域 $P_i$ 的 $S_i$ 号建筑上,且他的办公室位于 $Q_i$ 区域的 $T_i$ 号建筑上。

输出格式

输出仅为一行,包含一个整数,表示 $D_1 + D_2 + \dots + D_N$ 的最小值。

子任务

所有数据都保证:$P_i$ 和 $Q_i$ 为字符 “A” 和 “B” 中的一个, $0 \leq S_i, T_i \leq 1000000000$,同一栋建筑内可能有超过 $1$ 间房子或办公室(或二者的组合,即房子或办公室的数量同时大于等于 $1$)。

  • 子任务 1 (8 分)
    • $K = 1$
    • $1 \leq N \leq 1000$
  • 子任务 2 (14 分)
    • $K = 1$
    • $1 \leq N \leq 100000$
  • 子任务 3 (9 分)
    • $K = 2$
    • $1 \leq N \leq 100$
  • 子任务 4 (32 分)
    • $K = 2$
    • $1 \leq N \leq 1000$
  • 子任务 5 (37 分)
    • $K = 2$
    • $1 \leq N \leq 100000$

分析1

我们先考虑\(m=1\)的时候的做法

对于子任务1,首先把所有的坐标离散出来,最多有\(2n\)

然后暴力枚举\(2n\)的坐标的位置造桥,再\(O(n)\)计算代价

时间复杂度\(O(2n^2)\),期望得分8分

紧接着考虑优化,我们发现,不需要每次花费\(O(n)\)来计算代价

我们只需要把所有的区间离散到坐标上,然后顺着扫一遍,直接记录一下,每次就可以\(O(1)\)修改得到代价

时间复杂度\(O(2n)\),期望得分22分

#include<cstdio>  
#include<iostream>  
#include<algorithm>  
#include<cstdlib>  
#include<cstring>
#include<string>
#include<climits>
#include<vector>
#include<cmath>
#include<map>
#include<queue>
#define LL long long
#define pii pair<int,int>
#define mp make_pair
 
using namespace std;
 
inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
  return *p1++;
}
 
inline void read(int &x){
  char c=nc();int b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
 
inline void read(LL &x){
  char c=nc();LL b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
 
inline int read(char *s)
{
	char c=nc();int len=1;
	for(;!(c>='a' && c<='z');c=nc()) if (c==EOF) return 0;
	for(;(c>='a' && c<='z');s[len++]=c,c=nc());
	s[len++]='\0';
	return len;
}
 
inline void read(char &x){
  for (x=nc();!(x>='A' || x<='B');x=nc());
}
 
int wt,ss[19];
inline void print(int x){
	if (x<0) x=-x,putchar('-'); 
	if (!x) putchar(48); else {
	for (wt=0;x;ss[++wt]=x%10,x/=10);
	for (;wt;putchar(ss[wt]+48),wt--);}
}
inline void print(LL x){
	if (x<0) x=-x,putchar('-');
	if (!x) putchar(48); else {for (wt=0;x;ss[++wt]=x%10,x/=10);for (;wt;putchar(ss[wt]+48),wt--);}
}

int n,m,k;
LL b[2000010];
LL c[2000010],d[2000010];
struct data
{
	int xa,xb;
	LL ya,yb;
}a[1000010];

int Hash(int x)
{
    return lower_bound(b+1,b+1+k,x)-b;
}

int main()
{
	read(m);read(n);
	char ch;k=0;
	for (int i=1;i<=n;i++)
	{
		read(ch),read(a[i].ya),a[i].xa=ch-'A'+1,b[++k]=a[i].ya,
		read(ch),read(a[i].yb),a[i].xb=ch-'A'+1,b[++k]=a[i].yb;
		if (a[i].ya>a[i].yb) swap(a[i].ya,a[i].yb),swap(a[i].xa,a[i].xb);
	}
	sort(b+1,b+1+k);
	k=unique(b+1,b+1+k)-b-1;
	if (n<=1000 && m==1)
	{
		LL res=(LL)(1e17),s;
		for (int i=1;i<=k;i++)
		{
			s=0;
			for (int j=1;j<=n;j++)
				if (a[j].xa==a[j].xb) s+=a[j].yb-a[j].ya;
				else if (a[j].ya<=b[i] && a[j].yb>=b[i]) s+=a[j].yb-a[j].ya+1LL;
				else if (b[i]<a[j].ya) s+=a[j].yb-a[j].ya+1+(a[j].ya-b[i])*2LL;
				else s+=a[j].yb-a[j].ya+1LL+(b[i]-a[j].yb)*2LL;
			res=min(res,s);
		}
		print(res),puts("");
		return 0;
	}
	if (m==1 && n<=100000)
	{
		LL s=0,p=0,q=0;
		memset(c,0,sizeof(c));
		memset(d,0,sizeof(d));
		for (int i=1;i<=n;i++)
			if (a[i].xa==a[i].xb) s+=a[i].yb-a[i].ya;
			else p++,c[Hash(a[i].ya)]++,d[Hash(a[i].yb)]++,s+=a[i].yb-a[i].ya+1+(a[i].ya-b[1])*2;
		LL res=s;p-=c[1],q+=d[1];
		for (int i=2;i<=k;i++)
		{
			s-=2*p*(b[i]-b[i-1]);p-=c[i];
			s+=2*q*(b[i]-b[i-1]);q+=d[i];
			res=min(res,s);
		}
		print(res),puts("");
		return 0;
	}
	return 0;
}

分析2

然后在思考\(m=2\)做法的时候,猛然醒悟...自己原来是个煞笔...

我们在回过头去看\(m=1\)的时候

我们发现,答案由两部分组成,当两个建筑位于同一侧的时候\(ans1=|S_i-T_i|\),显然\(ans1\)是固定的

而第二部分的答案\(ans2=\sum |P_i-x|+|T_i-x|+1\)\(x\)表示桥的位置,显然当\(x\)集合\({S_i,T_i}\)的中位数的时候\(ans2\)取得最小值

时间复杂度\(O(nlogn)\),期望得分22

#include<cstdio>  
#include<iostream>  
#include<algorithm>  
#include<cstdlib>  
#include<cstring>
#include<string>
#include<climits>
#include<vector>
#include<cmath>
#include<map>
#include<queue>
#define LL long long
#define pii pair<int,int>
#define mp make_pair
 
using namespace std;
 
inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
  return *p1++;
}
 
inline void read(int &x){
  char c=nc();int b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
 
inline void read(LL &x){
  char c=nc();LL b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
 
inline int read(char *s)
{
	char c=nc();int len=1;
	for(;!(c>='a' && c<='z');c=nc()) if (c==EOF) return 0;
	for(;(c>='a' && c<='z');s[len++]=c,c=nc());
	s[len++]='\0';
	return len;
}
 
inline void read(char &x){
  for (x=nc();!(x>='A' || x<='B');x=nc());
}
 
int wt,ss[19];
inline void print(int x){
	if (x<0) x=-x,putchar('-'); 
	if (!x) putchar(48); else {
	for (wt=0;x;ss[++wt]=x%10,x/=10);
	for (;wt;putchar(ss[wt]+48),wt--);}
}
inline void print(LL x){
	if (x<0) x=-x,putchar('-');
	if (!x) putchar(48); else {for (wt=0;x;ss[++wt]=x%10,x/=10);for (;wt;putchar(ss[wt]+48),wt--);}
}

int n,m,k;
LL b[2000010];
LL c[2000010],d[2000010];
struct data
{
	int xa,xb;
	LL ya,yb;
}a[1000010];

int main()
{
	read(m);read(n);
	char ch;k=0;
	for (int i=1;i<=n;i++)
	{
		read(ch),read(a[i].ya),a[i].xa=ch-'A'+1;
		read(ch),read(a[i].yb),a[i].xb=ch-'A'+1;
		if (a[i].xa!=a[i].xb) b[++k]=a[i].ya,b[++k]=a[i].yb;
		if (a[i].ya>a[i].yb) swap(a[i].ya,a[i].yb),swap(a[i].xa,a[i].xb);
	}
	sort(b+1,b+1+k);
	if (m==1)
	{
		LL x=b[k/2+1],res=0;
		for (int i=1;i<=n;i++)
			if (a[i].xa==a[i].xb) res+=a[i].yb-a[i].ya;
			else if (a[i].ya<=x && a[i].yb>=x) res+=a[i].yb-a[i].ya+1;
			else if (a[i].ya>x) res+=a[i].yb-a[i].ya+1+2*(a[i].ya-x);
			else res+=a[i].yb-a[i].ya+1+2*(x-a[i].yb);
		print(res),puts("");
	}
	return 0;
}

分析3

考虑\(m=2\)的做法

对于子任务3,同上述一样离散以后,暴力枚举两座桥的位置,然后花费\(O(n)\)计算代价

时间复杂度\(O(4n^3)\),期望得分9分

继续考虑,我们可以发现,当有两座桥的时候,所有的居民唯一需要做的就是,找一座代价小的桥

而对于每一个居民而言,代价\(ans=|P_i-x|+|T_i-x|\)\(x\)为桥的位置,即如下图

那么,增加相同的距离,也就是说,我们需要比较\(\frac {P_i+T_i}{2}\)距离哪个比较近即可

这样,我们如果把所有的居民按照\(\frac {P_i+T_i}{2}\)排序的话,我们可以把所有的居民分成连续的两段,前一段使用前一座桥,后一段时候后一座桥

那么,我们把\(m=2\)问题,规约成了两个\(m=1\)问题

枚举断点,暴力的做\(m=1\)问题即可

时间复杂度\(O(n^2logn)\),期望得分41分,合并第一部分,可以得到63分,是一个很可观的分数了

#include<cstdio>  
#include<iostream>  
#include<algorithm>  
#include<cstdlib>  
#include<cstring>
#include<string>
#include<climits>
#include<vector>
#include<cmath>
#include<map>
#include<queue>
#define LL long long
#define pii pair<int,int>
#define mp make_pair
 
using namespace std;
 
inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
  return *p1++;
}
 
inline void read(int &x){
  char c=nc();int b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
 
inline void read(LL &x){
  char c=nc();LL b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
 
inline int read(char *s)
{
	char c=nc();int len=1;
	for(;!(c>='a' && c<='z');c=nc()) if (c==EOF) return 0;
	for(;(c>='a' && c<='z');s[len++]=c,c=nc());
	s[len++]='\0';
	return len;
}
 
inline void read(char &x){
  for (x=nc();!(x>='A' || x<='B');x=nc());
}
 
int wt,ss[19];
inline void print(int x){
	if (x<0) x=-x,putchar('-'); 
	if (!x) putchar(48); else {
	for (wt=0;x;ss[++wt]=x%10,x/=10);
	for (;wt;putchar(ss[wt]+48),wt--);}
}
inline void print(LL x){
	if (x<0) x=-x,putchar('-');
	if (!x) putchar(48); else {for (wt=0;x;ss[++wt]=x%10,x/=10);for (;wt;putchar(ss[wt]+48),wt--);}
}

int n,m,k;
LL b[2000010];
vector<int> e;
LL c[2000010],d[2000010];
struct data
{
	int xa,xb;
	LL ya,yb;
}a[1000010];

bool cmp(data x,data y)
{
	return x.ya+x.yb<y.ya+y.yb;
}

int main()
{
	read(m);read(n);
	char ch;k=0;LL S=0;int x=0;
	for (int i=1;i<=n;i++)
	{
		x++;
		read(ch),read(a[x].ya),a[x].xa=ch-'A'+1;
		read(ch),read(a[x].yb),a[x].xb=ch-'A'+1;
		if (a[x].xa==a[x].xb) {S+=abs(a[x].ya-a[x].yb),x--;continue;}
		b[++k]=a[x].ya,b[++k]=a[x].yb;
		if (a[x].ya>a[x].yb) swap(a[x].ya,a[x].yb),swap(a[x].xa,a[x].xb);
	}
	n=x;
	sort(b+1,b+1+k);
	if (m==2 && n<=1000)
	{
		sort(a+1,a+1+n,cmp);
		LL res=(LL)1e17,s,y;
		for (int x=1;x<=n+1;x++)
		{
			s=0;
			if (x>1)
			{
				e.clear();
				for (int i=1;i<x;i++)
					e.push_back(a[i].ya),e.push_back(a[i].yb);
				sort(e.begin(),e.end());
				y=e[e.size()/2];
				for (int i=1;i<x;i++)
					s+=abs(a[i].ya-y)+abs(a[i].yb-y)+1;
			}
			if (x<=n)
			{
				e.clear();
				for (int i=x;i<=n;i++)
					e.push_back(a[i].ya),e.push_back(a[i].yb);
				sort(e.begin(),e.end());
				y=e[e.size()/2];
				for (int i=x;i<=n;i++)
					s+=abs(a[i].ya-y)+abs(a[i].yb-y)+1;
			}
			res=min(res,s);
		}
		print(res+S),puts("");
	}
	return 0;
}

分析4

继续考虑\(m=2\)的满分做法

我们得想办法把找中位数和算\(\sum |P_i-x|+|T_i-x|+1\)加速

我们每次会加入两个数字,或者删除两个数字,需要动态的维护中位数和上面的绝对值之和

用线段树来维护,很容易维护出中位数

那么绝对值怎么处理?显然,把绝对值拆开,因为,维护中位数的时候,顺便记录下中位数之前的和,然后就可以做啦

时间复杂度\(O(nlogn)\),期望得分100


满分程序

我的线段树写的比较丑QAQ

#include<cstdio>  
#include<iostream>  
#include<algorithm>  
#include<cstdlib>  
#include<cstring>
#include<string>
#include<climits>
#include<vector>
#include<cmath>
#include<map>
#include<queue>
#define LL long long
#define pii pair<int,int>
#define mp make_pair
 
using namespace std;
 
inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
  return *p1++;
}
 
inline void read(int &x){
  char c=nc();int b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
 
inline void read(LL &x){
  char c=nc();LL b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}
 
inline int read(char *s)
{
	char c=nc();int len=1;
	for(;!(c>='a' && c<='z');c=nc()) if (c==EOF) return 0;
	for(;(c>='a' && c<='z');s[len++]=c,c=nc());
	s[len++]='\0';
	return len;
}
 
inline void read(char &x){
  for (x=nc();!(x>='A' || x<='B');x=nc());
}
 
int wt,ss[19];
inline void print(int x){
	if (x<0) x=-x,putchar('-'); 
	if (!x) putchar(48); else {
	for (wt=0;x;ss[++wt]=x%10,x/=10);
	for (;wt;putchar(ss[wt]+48),wt--);}
}
inline void print(LL x){
	if (x<0) x=-x,putchar('-');
	if (!x) putchar(48); else {for (wt=0;x;ss[++wt]=x%10,x/=10);for (;wt;putchar(ss[wt]+48),wt--);}
}

int n,m,k;
LL b[200010];
LL c[200010],d[200010];
struct data
{
	int xa,xb;
	LL ya,yb;
}a[100010];
struct ST
{
	int sum;LL x;
}s1[800010],s2[800010];

int Hash(int x)
{
    return lower_bound(b+1,b+1+k,x)-b;
}

void change1(int q,int l,int r,int x,int y,LL z)
{
	if (l==r) {s1[x].sum+=y;s1[x].x+=z;return ;}
	int mid=l+r>>1;
	if (q<=mid) change1(q,l,mid,x<<1,y,z);else change1(q,mid+1,r,x<<1|1,y,z);
	s1[x].sum=s1[x<<1].sum+s1[x<<1|1].sum;
	s1[x].x=s1[x<<1].x+s1[x<<1|1].x;
}

void change2(int q,int l,int r,int x,int y,LL z)
{
	if (l==r) {s2[x].sum+=y;s2[x].x+=z;return ;}
	int mid=l+r>>1;
	if (q<=mid) change2(q,l,mid,x<<1,y,z);else change2(q,mid+1,r,x<<1|1,y,z);
	s2[x].sum=s2[x<<1].sum+s2[x<<1|1].sum;
	s2[x].x=s2[x<<1].x+s2[x<<1|1].x;
}

int query1(int q,int l,int r,int x)
{
	if (l==r) return b[l];
	int mid=l+r>>1;
	if (s1[x<<1].sum>=q) return query1(q,l,mid,x<<1);
	return query1(q-s1[x<<1].sum,mid+1,r,x<<1|1);
}

int query2(int q,int l,int r,int x)
{
	if (l==r) return b[l];
	int mid=l+r>>1;
	if (s2[x<<1].sum>=q) return query2(q,l,mid,x<<1);
	return query2(q-s2[x<<1].sum,mid+1,r,x<<1|1);
}

LL sum1(int q,int l,int r,int x)
{
	if (l==r) return s1[x].x;
	int mid=l+r>>1;
	if (s1[x<<1].sum>=q) return sum1(q,l,mid,x<<1);
	return s1[x<<1].x+sum1(q-s1[x<<1].sum,mid+1,r,x<<1|1);
}

LL sum2(int q,int l,int r,int x)
{
	if (l==r) return s2[x].x;
	int mid=l+r>>1;
	if (s2[x<<1].sum>=q) return sum2(q,l,mid,x<<1);
	return s2[x<<1].x+sum2(q-s2[x<<1].sum,mid+1,r,x<<1|1);
}

LL num1(int q,int l,int r,int x)
{
	if (l==r) return s1[x].sum;
	int mid=l+r>>1;
	if (s1[x<<1].sum>=q) return num1(q,l,mid,x<<1);
	return s1[x<<1].sum+num1(q-s1[x<<1].sum,mid+1,r,x<<1|1);
}

LL num2(int q,int l,int r,int x)
{
	if (l==r) return s2[x].sum;
	int mid=l+r>>1;
	if (s2[x<<1].sum>=q) return num2(q,l,mid,x<<1);
	return s2[x<<1].sum+num2(q-s2[x<<1].sum,mid+1,r,x<<1|1);
}

bool cmp(data x,data y)
{
	return x.ya+x.yb<y.ya+y.yb;
}

int main()
{
	read(m);read(n);
	char ch;k=0;LL S=0;int x=0;
	for (int i=1;i<=n;i++)
	{
		x++;
		read(ch),read(a[x].ya),a[x].xa=ch-'A'+1;
		read(ch),read(a[x].yb),a[x].xb=ch-'A'+1;
		if (a[x].xa==a[x].xb) {S+=abs(a[x].ya-a[x].yb),x--;continue;}
		b[++k]=a[x].ya,b[++k]=a[x].yb;
		if (a[x].ya>a[x].yb) swap(a[x].ya,a[x].yb),swap(a[x].xa,a[x].xb);
	}
	n=x;
	sort(b+1,b+1+k);
	if (m==1)
	{
		LL x=b[k/2+1],res=0;
		for (int i=1;i<=n;i++)
			if (a[i].ya<=x && a[i].yb>=x) res+=a[i].yb-a[i].ya+1;
			else if (a[i].ya>x) res+=a[i].yb-a[i].ya+1+2*(a[i].ya-x);
			else res+=a[i].yb-a[i].ya+1+2*(x-a[i].yb);
		print(res+S),puts("");
	}
	if (m==2)
	{
		memset(s1,0,sizeof(s1));
		memset(s2,0,sizeof(s2));
		sort(a+1,a+1+n,cmp);
		LL res=(LL)1e17,s,y,z,t,p=0,q=0;
		for (int i=1;i<=n;i++)
			change2(Hash(a[i].ya),1,k,1,1,a[i].ya),change2(Hash(a[i].yb),1,k,1,1,a[i].yb),q+=2LL;
		for (int x=1;x<=n+1;x++)
		{
			s=0;
			if (x>1)
			{
				y=query1(p/2+1,1,k,1);
				z=sum1(p/2+1,1,k,1);
				t=num1(p/2+1,1,k,1);
				s+=y*t-z+(s1[1].x-z)-y*(p-t)+(LL)(x-1);
			}
			if (x<=n)
			{
				y=query2(q/2+1,1,k,1);
				z=sum2(q/2+1,1,k,1);
				t=num2(q/2+1,1,k,1);
				s+=y*t-z+(s2[1].x-z)-y*(q-t)+(LL)(n-x+1);
			}
			if (x<=n)
			{
				change1(Hash(a[x].ya),1,k,1,1,a[x].ya),change1(Hash(a[x].yb),1,k,1,1,a[x].yb);p+=2LL;
				change2(Hash(a[x].ya),1,k,1,-1,-a[x].ya),change2(Hash(a[x].yb),1,k,1,-1,-a[x].yb);q-=2LL;
			}
			res=min(res,s);
		}
		print(res+S),puts("");
	}
	return 0;
}
posted @ 2017-04-20 15:38  Xiejiadong  阅读(334)  评论(0编辑  收藏  举报
levels of contents