杭电5101——二分依旧浪(upper_bound和lower_bound)


奋斗题意大概:首先输入T,有T组测试数据。然后下一行输入n,m,表示有几个班,m表示自己的IQ值。后面跟上n行,每行第一个数代表这个班里有几个人,后面

跟上班里人的IQ值。     所要做的就是在两个不同班里找两个同学,要这两个的IQ值和大于自己的IQ值,问有多少种搭配?



这里其实有个小思想:       不同两个班级的两个同学IQ值和大于m的方式数=所有同学搭配方式数同班两个同学IQ值大于m的方式数

再介绍两个逆天的函数:upper_bound和lower_bound

  ForwardIter lower_bound(ForwardIter first, ForwardIter last,const _Tp& val)算法返回一个非递减序列[first, last)中

的第一个大于等于值val的位置。

      ForwardIter upper_bound(ForwardIter first, ForwardIter last, const _Tp& val)算法返回一个非递减序列[first, last)中

第一个大于val的位置。

      lower_bound和upper_bound如下图所示:


#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
int main()
{
	int T;
	int n,m,i,j,k,l,le,ri,mid,a;
	__int64 res,ans,ant;
	scanf("%d",&T);
	while(T--)
	{
		__int64 node[100100];
		scanf("%d%d",&n,&m);
		l=0;res=0;ant=0;
		for(j=1;j<=n;j++)
		{
			scanf("%d",&k);
			__int64 p[101];
			for(i=0;i<k;i++)
			{
				scanf("%I64d",&p[i]); 
				node[l++]=p[i];       //把全部人的IQ值全放到个数组里,不分班级
			}
			sort(p,p+k);//此时的p内都是同一个班内的,在这个循环里就求出同班内,符合情况的两个小伙伴的方式数
			for(i=k-1;i>=1;i--)
			{
				 ans=i-(upper_bound(p,p+k,m-p[i])-p);
				 //当然如果要换成传统的二分也是可以的,但是虽然直接用函数非常简便
				 //,但是用这个函数的局限性太大,所以还是要懂他的原理,如下:
				/************
				ri=i-1;le=0;
				a=-999999;
				while(ri>=le)
				{
					mid=(ri+le)/2;
					if(p[mid]>(m-p[i]))
					{
						a=mid;
						ri=mid-1;
					}
					else
						le=mid+1;
				}
				ans=i-a;***********/
				if(ans>0&&a>=0)
					ant+=ans;
				else
					break;
			}
		}
		sort(node,node+l);
		for(i=l-1;i>=1;i--)
		{
			ans=i-(upper_bound(node,node+l,m-node[i])-node);
			if(ans>0)
				res+=ans;//求出总的方式数
			else
				break;
		}
		res-=ant;//总的不分班方式数-同班内的方式数=题意方式数
		printf("%I64d\n",res);
	}
	return 0;
}

 







  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值