「杂题乱刷」AT_abc360_d

题目链接

AT_abc360_d (luogu)

AT_abc360_d (atcoder)

解题思路

一个性质是，往左边走的蚂蚁无论怎么样都追不到左边的蚂蚁，而往右边走的蚂蚁无论怎么样都追不上右边的蚂蚁。

因此我们考虑将蚂蚁分为往左往右走的两堆。

发现对于每个蚂蚁都能走过一段区间，因此直接二分将右端点减去左端点即可。

时间复杂度 \(O(n \log_2 n)\)。

参考代码

点击查看代码

/*
Tips:
你数组开小了吗？
你MLE了吗？
你觉得是贪心，是不是该想想dp？
一个小时没调出来，是不是该考虑换题？
打 cf 不要用 umap！！！

记住，rating 是身外之物。

该冲正解时冲正解！

Problem:

算法：

思路：

*/
#include<bits/stdc++.h>
using namespace std;
//#define map unordered_map
#define re register
#define ll long long
#define forl(i,a,b) for(re ll i=a;i<=b;i++)
#define forr(i,a,b) for(re ll i=a;i>=b;i--)
#define forll(i,a,b,c) for(re ll i=a;i<=b;i+=c)
#define forrr(i,a,b,c) for(re ll i=a;i>=b;i-=c)
#define lc(x) x<<1
#define rc(x) x<<1|1
#define mid ((l+r)>>1)
#define cin(x) scanf("%lld",&x)
#define cout(x) printf("%lld",x)
#define lowbit(x) (x&-x)
#define pb push_back
#define pf push_front
#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
#define endl '\n'
#define QwQ return 0;
#define db long double
#define ull unsigned long long
#define lcm(x,y) x/__gcd(x,y)*y
#define Sum(x,y) 1ll*(x+y)*(y-x+1)/2
#define aty cout<<"Yes\n";
#define atn cout<<"No\n";
#define cfy cout<<"YES\n";
#define cfn cout<<"NO\n";
#define xxy cout<<"yes\n";
#define xxn cout<<"no\n";
#define printcf(x) x?cout<<"YES\n":cout<<"NO\n";
#define printat(x) x?cout<<"Yes\n":cout<<"No\n";
#define printxx(x) x?cout<<"yes\n":cout<<"no\n";
ll t;
ll n,m;
string s;
ll a[200010];
ll b[200010],c[200010],k1,k2;
ll ans;

void solve()
{
	cin>>n>>m>>s;
	forl(i,1,n)
		cin>>a[i];
	forl(i,1,n)
	{
		if(s[i-1]=='0')
			c[++k2]=a[i];
		else
			b[++k1]=a[i];
	}
	sort(b+1,b+1+k1);
	sort(c+1,c+1+k2);
	if(k1==0 || k2==0)
	{
		cout<<0<<endl;
		return ;
	}
	forl(i,1,k2)
	{
		ll L=1,R=k1;
		while(L<R)
		{
			ll Mid=(L+R)/2;
			if(c[i]-b[Mid]<=2*m)
				R=Mid;
			else
				L=Mid+1;
		}
		if(c[i]-b[L]>m*2)
			continue;
		ll sum1=L;
		L=1,R=k1;
		while(L<R)
		{
			ll Mid=(L+R+1)/2;
			if(b[Mid]<=c[i])
				L=Mid;
			else
				R=Mid-1; 
		}
		if(b[L]<=c[i])
			ans+=L-sum1+1; 
	}
	cout<<ans<<endl;
}
int main()
{
	IOS;
	t=1;
 //	cin>>t;
	while(t--)
		solve();
    /******************/
	/*while(L<q[i].l) */
	/*    del(a[L++]);*/
	/*while(L>q[i].l) */
	/*    add(a[--L]);*/
	/*while(R<q[i].r) */
	/*	  add(a[++R]);*/
	/*while(R>q[i].r) */
	/*    del(a[R--]);*/
    /******************/
	QwQ;
}