三足鼎立(二分)"/>
pta三足鼎立(二分)
L2-2 三足鼎立
当三个国家中的任何两国实力之和都大于第三国的时候,这三个国家互相结盟就呈“三足鼎立”之势,这种状态是最稳定的。
现已知本国的实力值,又给出 n 个其他国家的实力值。我们需要从这 n 个国家中找 2 个结盟,以成三足鼎立。有多少种选择呢?
输入格式:
输入首先在第一行给出 2 个正整数 n(2≤n≤105)和 P(≤109),分别为其他国家的个数、以及本国的实力值。随后一行给出 n 个正整数,表示n 个其他国家的实力值。每个数值不超过 109,数字间以空格分隔。
输出格式:
在一行中输出本国结盟选择的个数。
输入样例:
7 30
42 16 2 51 92 27 35
输出样例:
9
样例解释:
能联合的另外 2 个国家的 9 种选择分别为:
{16, 27}, {16, 35}, {16, 42}, {27, 35}, {27, 42}, {27, 51}, {35, 42}, {35, 51}, {42, 51}。
本题的基本思维很简单,主要是会超时
本题暴力求解为O方
关键就是去除无用数据的出现
根据题意可知 a+b>c c+b>a a+c>b
根据上列表达式可知,a+b>c c>a-b c>b-a可知道c的取值范围abs(a-b)<c<a+b
于是我们可以先把国家数据进行排列,随后用二分查找范围end-beg+1就是符合条件的国家啦
+1是因为beg本身也满足条件于是要加上1
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.StreamTokenizer;
import java.util.Arrays;public class Main三足鼎立 {public static void main(String[] args) throws IOException {StreamTokenizer x = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));x.nextToken();int n=(int)x.nval;x.nextToken();int m=(int)x.nval;long l=0;int aa[]=new int[n];for(int i=0;i<n;i++) {x.nextToken();aa[i]=(int)x.nval;}Arrays.sort(aa);//c>a-b c>b-a c<a+bfor(int i=0;i<n;i++) {int p=aa[i];long sun1=m+p,sun2=Math.abs(m-p),a,b;int beg=i+1,end=n-1,mod;//从i+1开始去重为目的while(beg<=end) {mod=(beg+end)/2;if(aa[mod]>=sun1)end=mod-1;elsebeg=mod+1;}a=end;beg=i+1;end=n-1;while(beg<=end) {mod=(beg+end)/2;if(aa[mod]<=sun2)beg=mod+1;elseend=mod-1;}b=beg;l+=a-b+1;}out.println(l);out.flush();}
}
更多推荐
pta三足鼎立(二分)
发布评论