[BZOJ2244]:拦截导弹(DP+CDQ分治+树状数组)

题目传送门


题目描述

某国为了防御敌国的导弹袭击,发展出一种导弹拦截系统。但是这种导弹拦截系统有一个缺陷:虽然它的第一发炮弹能够到达任意的高度、并且能够拦截任意速度的导弹,但是以后每一发炮弹都不能高于前一发的高度,其拦截的导弹的飞行速度也不能大于前一发。某天,雷达捕捉到敌国的导弹来袭。由于该系统还在试用阶段,所以只有一套系统,因此有可能不能拦截所有的导弹。
在不能拦截所有的导弹的情况下,我们当然要选择使国家损失最小、也就是拦截导弹的数量最多的方案。但是拦截导弹数量的最多的方案有可能有多个,如果有多个最优方案,那么我们会随机选取一个作为最终的拦截导弹行动蓝图。
我方间谍已经获取了所有敌军导弹的高度和速度,你的任务是计算出在执行上述决策时,每枚导弹被拦截掉的概率。


输入格式

第一行包含一个正整数$n$,表示敌军导弹数量;
下面行按顺序给出了敌军所有导弹信息:
第$i+1$行包含$2$个正整数$h_i$和$v_i$,分别表示第$i$枚导弹的高度和速度。


输出格式

输出包含两行。
第一行为一个正整数,表示最多能拦截掉的导弹数量;
第二行包含$n$个$0$到$1$之间的实数,第$i$个数字表示第$i$枚导弹被拦截掉的概率(你可以保留任意多位有效数字)。


样例

样例输入

4
3 30
4 40
6 60
3 30

样例输出

2
0.33333 0.33333 0.33333 1.00000


数据范围与提示

对于$100%$的数据,$1\leqslant n\leqslant 5\times 10^4$,$1\leqslant h_i$ ,$v_i\leqslant 10^9$;
均匀分布着约$30%$的数据,所有$v_i$均相等。
均匀分布着约$50%$的数据,满足$1\leqslant h_i,v_i\leqslant 1,000$。


题解

这道题就是求一个三维的最长非上升子序列的长度($LDS$),肯定要用$CDQ$分治啦~

利用中序遍历的思想来进行$CDQ$分治,先处理左区间,再处理当前节点的值,最后处理右区间。

设$f_{1_i}$表示正序时$LDS$的长度,$g_{1_i}$表示个数,$f_{2_i}$和$g_{2_i}$则是逆序。

处理跟$CDQ$分治一样。

下面来讲一下如何统计答案:

  第一问的答案就是$\max(f_{1_i})$。

  第二问稍繁琐,考虑只有当$f_{1_i}+f_{2_i}-1=\max(f_{1_i})$时(减去重复计算的节点$i$),给节点才能成为$LDS$上的点,总方案数即为$g_{1_i}\times g_{2_i}$,答案即为$\frac{g_{1_i}\times g_{2_i}}{\sum \limits_{i=1}^{n}g_{1_i}(f_{1_i}=\max)}$。


代码时刻

#include<bits/stdc++.h>
using namespace std;
struct rec
{
	int d;
	int h;
	int v;
	pair<pair<int,double>,pair<int,double> >dp;
}e[50001];
int n;
int flag[50001],hmax,vmax;
int trmax[200001];
int ans;
double sum;
pair<int,double> tr[200000];//树状数组
bool cmpd(rec a,rec b){return a.d<b.d;}
bool cmph1(rec a,rec b){return a.h>b.h;}
bool cmph2(rec a,rec b){return a.h<b.h;}
bool cmpv(rec a,rec b){return a.v<b.v;}
int lowbit(int x){return x&-x;}
void mem(int x)//清空
{
	for(int i=x;i<=n;i+=lowbit(i))
		tr[i]=make_pair(0,0.0);
}
void add(int x,int l,double r)//插入
{
	for(int i=x;i<=n;i+=lowbit(i))
		if(tr[i].first<l)tr[i]=make_pair(l,r);
		else if(tr[i].first==l)tr[i].second+=r;
}
pair<int,double> ask(int x)//查询
{
	pair<int,double> res;
	for(int i=x;i;i-=lowbit(i))
		if(tr[i].first>res.first)res=tr[i];
		else if(tr[i].first==res.first)res.second+=tr[i].second;
	return res;
}
void cdq1(int l,int r)//正序CDQ
{
	if(l==r)return;
	int mid=(l+r)>>1;
	cdq1(l,mid);
	sort(e+l,e+mid+1,cmph1);
	sort(e+mid+1,e+r+1,cmph1);
	int i=mid+1,j=l;
	for(;i<=r;i++)
	{
		while(j<=mid&&e[i].h<=e[j].h){add(vmax-e[j].v,e[j].dp.first.first,e[j].dp.first.second);j++;}
		pair<int,double> frec=ask(vmax-e[i].v);
		if(e[i].dp.first.first<=frec.first)
		{
			e[i].dp.first.first=frec.first+1;
			e[i].dp.first.second=frec.second;
		}
		else if(e[i].dp.first.first==frec.first+1)e[i].dp.first.second+=frec.second;
	}j--;
	for(;j>=l;j--)mem(vmax-e[j].v);
	sort(e+l,e+r+1,cmpd);
	cdq1(mid+1,r);
}
void cdq2(int l,int r)//逆序CDQ
{
	if(l==r)return;
	int mid=(l+r)>>1;
	cdq2(mid+1,r);
	sort(e+l,e+mid+1,cmph2);
	sort(e+mid+1,e+r+1,cmph2);
	int i=l,j=mid+1;
	for(;i<=mid;i++)
	{
		while(j<=r&&e[i].h>=e[j].h){add(e[j].v,e[j].dp.second.first,e[j].dp.second.second);j++;}
		pair<int,double> frec=ask(e[i].v);
		if(e[i].dp.second.first<=frec.first)
		{
			e[i].dp.second.first=frec.first+1;
			e[i].dp.second.second=frec.second;
		}
		else if(e[i].dp.second.first==frec.first+1)e[i].dp.second.second+=frec.second;
	}j--;
	for(;j>mid;j--)mem(e[j].v);
	sort(e+l,e+r+1,cmpd);
	cdq2(l,mid);
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		e[i].d=i;
		scanf("%d%d",&e[i].h,&e[i].v);
		e[i].dp.first.first=e[i].dp.second.first=1;
		e[i].dp.first.second=e[i].dp.second.second=1.0;
	}
	sort(e+1,e+n+1,cmph2);
	for(int i=1;i<=n;i++)//预处理h
	{
		if(e[i].h!=e[i-1].h)flag[0]++;
		flag[i]=flag[0];
	}
	for(int i=1;i<=n;i++)
		e[i].h=flag[i];
	flag[0]=0;
	sort(e+1,e+n+1,cmpv);
	for(int i=1;i<=n;i++)//预处理v
	{
		if(e[i].v!=e[i-1].v)flag[0]++;
		flag[i]=flag[0];
	}
	for(int i=1;i<=n;i++)
		e[i].v=flag[i];
	vmax=e[n].v+1;
	sort(e+1,e+n+1,cmpd);
	cdq1(1,n);
	for(int i=1;i<=n;i++)//第一问
		ans=max(ans,e[i].dp.first.first);
	cdq2(1,n);
	for(int i=1;i<=n;i++)
		if(e[i].dp.first.first+e[i].dp.second.first==ans+1)
			sum+=e[i].dp.first.second*e[i].dp.second.second;
	sum/=(double)ans;
	printf("%d\n",ans);
	for(int i=1;i<=n;i++)//第二问
	{
		if(e[i].dp.first.first+e[i].dp.second.first==ans+1)
			printf("%.7lf ",e[i].dp.first.second*e[i].dp.second.second/sum);
		else printf("0.0000000 ");
	}
	return 0;
}

rp++

posted @ 2019-08-02 17:15  HEOI-动动  阅读(239)  评论(0编辑  收藏  举报