Codeforces Round #624 (Div. 3) F. Moving Points(树状数组+思维)

题目链接
在这里插入图片描述
在这里插入图片描述
思路:一开始一直想不通为什么这道题过的人那么多,后来才发现自己是傻逼,题目明明说了t不一定是整数,还在算两点相遇的时间。。。(好气啊,不然就可以AK div3了)。
如果xi<xj并且vi<vj的话那么i和j永远不能相遇,贡献为0,所以对于i来说要求的就是位置在i之前,速度还比i大的点,树状数组维护一下点的个数和前缀和就行了。。。

#include <bits/stdc++.h>
#define ll long long
#define lowbit(i)  (i)&(-i)
using namespace std;
const int maxn=2e5+1; 
ll c[maxn][2],ans,b[maxn];
struct node{
	ll x,v;
}s[maxn];
bool cmp(const node &a,const node &b)
{
	return a.x<b.x;
}
void add(int x,int val)
{
	while(x<maxn) c[x][0]++,c[x][1]+=val,x+=lowbit(x);
}
ll query(int x,int k)
{
	ll res=0;
	while(x>0) res+=c[x][k],x-=lowbit(x);
	return res;
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;++i) scanf("%lld",&s[i].x);
	for(int i=1;i<=n;++i) scanf("%lld",&s[i].v),b[i]=s[i].v;
	sort(s+1,s+1+n,cmp);
	sort(b+1,b+1+n);
	int size=unique(b+1,b+1+n)-b-1;
	for(int i=1;i<=n;++i)
	{
		int pos=lower_bound(b+1,b+1+size,s[i].v)-b;
		ans+=query(pos,0)*s[i].x-query(pos,1);
		add(pos,s[i].x);
	}
	printf("%lld\n",ans);
}
发布了144 篇原创文章 · 获赞 0 · 访问量 4939

猜你喜欢

转载自blog.csdn.net/qq_42479630/article/details/104524346