90分,TLE了一个,求解
查看原帖
90分,TLE了一个,求解
53022
银杉水杉秃杉楼主2020/12/2 13:11
#include<bits/stdc++.h>
#define ll long long
#define pi pair<int,int>
using namespace std;
ll n,m,h,ans;
ll c[300010],s[300010];
struct node1
{
	ll x,id;
}a[300010];
struct node2
{
	ll l,r,id;
}b[300010];
pi p[300010]; 
bool cmp1(node1 a,node1 b)
{
	return a.x<b.x;
}
bool cmp2(pi a,pi b)
{
	return a.first<b.first;
}
bool cmp3(node2 a,node2 b)
{
	return a.l!=b.l?a.l<b.l:a.r<b.r;
}
void add(node1 a,node1 b)
{
	p[++h].first=min(a.id,b.id);
	p[h].second=max(a.id,b.id);
}
void Update(ll x,ll y)
{
	for (;x<=n;x+=(x&-x))
		c[x]+=y;
}
ll Query(ll x)
{
	ll sum=0;
	for (;x>=1;x-=(x&-x))
		sum+=c[x];
	return sum; 
}
int main()
{
	scanf("%lld%lld",&n,&m);
	for (int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i].x);
		a[i].id=i;
	}
	sort(a+1,a+n+1,cmp1);
	add(a[2],a[1]);
	add(a[n],a[n-1]);
	for (int i=2;i<n;i++)
	{
		if (abs(a[i].x-a[i-1].x)==abs(a[i+1].x-a[i].x))
		{
			add(a[i],a[i-1]);
			add(a[i+1],a[i]);
		}
		if (abs(a[i].x-a[i-1].x)<abs(a[i+1].x-a[i].x)) 
			add(a[i],a[i-1]);
		if (abs(a[i].x-a[i-1].x)>abs(a[i+1].x-a[i].x)) 
			add(a[i+1],a[i]);
	}
	sort(p+1,p+h+1,cmp2);
	for (int i=1;i<=m;i++)
	{
		scanf("%lld%lld",&b[i].l,&b[i].r);
		b[i].id=i;
	}
	sort(b+1,b+m+1,cmp3);
	int j=h;
	for (int i=m;i>=1;i--)
	{
		while (p[j].first>=b[i].l)
		{
			Update(p[j].second,1);
			j--;
		}
		s[b[i].id]=Query(b[i].r);
	}
	for (int i=1;i<=m;i++)
		ans+=s[i]*i;
	printf("%lld\n",ans);
	return 0;
} 
2020/12/2 13:11
加载中...