AcWing 1945. 奶牛棒球（枚举+二分）

Posted 2022-02-12 MangataTS

题目链接

https://www.acwing.com/problem/content/1947/

思路

因为第三头牛和第二头牛的间距是在[2Y-X,3Y-2x]以内的，所以我们可以对第三头牛进行二分搜索，然后将得到的左右区间长度的贡献加在ans上面，详情请看代码

代码

#include<bits/stdc++.h>
using namespace std;
//----------------自定义部分----------------
#define ll long long
#define mod 1000000007
#define endl "\\n"
#define PII pair<int,int>

int dx[4]=0,-1,0,1,dy[4]=-1,0,1,0;

ll ksm(ll a,ll b) 
	ll ans = 1;
	for(;b;b>>=1LL) 
		if(b & 1) ans = ans * a % mod;
		a = a * a % mod;
	
	return ans;


ll lowbit(ll x)return -x & x;

const int N = 2e6+10;
//----------------自定义部分----------------
int n,m,q,a[N];

int main()

	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	std::cout.tie(nullptr);
	cin>>n;
	for(int i = 1;i <= n; ++i) 
		cin>>a[i];
	
	int ans = 0;
	sort(a+1,a+1+n);
	for(int i = 1;i <= n - 2; ++i) 
		for(int j = i + 1;j <= n - 1; ++j) 
			int l = upper_bound(a+j+1,a+n+1,2*a[j]-a[i]-1) - a;
			int r = upper_bound(a+j+1,a+n+1,2*(a[j]-a[i]) + a[j]) - a;
			ans += r - l;
		
	
	cout<<ans<<endl;
	return 0;