笛卡尔树45分求调
查看原帖
笛卡尔树45分求调
753448
Alstar楼主2024/5/7 17:35

代码如下 红绿相加 错了3,4,6,7,8,10,12,16,18,19,20 看不出来问题

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int N=3e5+5;
int n,k,ans;
int b[N];
int rs[N],ls[N],sta[N],top,root;
struct str{
	int num,sum;
}a[N];

bool cmp(int x,int y){ return x>y; }
bool cmp2(str x,str y){ return x.sum>y.sum; }

int dp(int p)
{
	if(p==0)
		return 0;
	a[p].sum=max(dp(rs[p])+1,dp(ls[p])+1);
	return a[p].sum;
}


int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i].num;		
		if(a[i].num>a[root].num)
			root=i;
	}
	for(int i=1;i<=k;i++)
		cin>>b[i];
	sort(b+1,b+1+n,cmp);
	//构建笛卡尔树 
	for(int i=1;i<=n;i++)
	{
		int p=top;
		while(a[i].num>a[sta[p]].num && p>0) p--;
		if(p>0) rs[sta[p]]=i;
		if(p<top) ls[i]=sta[p+1];
		sta[++p]=i;
		top=p;
	}
	//dp
	dp(root);
	ans+=a[1].sum-1;
	sort(a+1,a+1+n,cmp2);
	top=1;
	for(int i=1;i<=n;i++)
	{
		while(b[top]>=a[i].num && top<=k)
		{
			ans+=a[i].sum-1;
			top++;
		}
	}
	cout<<ans;
	return 0;
}
/*
11 3
9 3 7 1 8 12 10 20 15 18 5
20 3 9

*/

2024/5/7 17:35
加载中...