51nod 1305 Pairwise Sum and Divide

最新推荐文章于 2019-09-03 21:21:28 发布

BBHHTT

最新推荐文章于 2019-09-03 21:21:28 发布

阅读量174

点赞数

分类专栏： 1级算法题

本文链接：https://blog.csdn.net/BBHHTT/article/details/80088573

版权

1级算法题专栏收录该内容

18 篇文章 0 订阅

订阅专栏

在51nod的1305题中，初试暴力方法导致超时。题解指出，只有少数情况下A[i]+A[j] >= A[i]*A[j]成立。当A[i]=A[j]=2时取1，A[i]、A[j]之一为1时也取1，A[i]=A[j]=1则取2。根据2和1的数量，可以通过组合数学计算符合条件的总数。对于1的情况，需要避免重复计算。

摘要由CSDN通过智能技术生成

1305 Pairwise Sum and Divide

题目来源： HackerRank

基准时间限制：1 秒空间限制：131072 KB 分值: 5 难度：1级算法题

关注

有这样一段程序，fun会对整数数组A进行求值，其中Floor表示向下取整：

fun(A)

sum = 0

for i = 1 to A.length

for j = i+1 to A.length

sum = sum + Floor((A[i]+A[j])/(A[i]*A[j]))

return sum

给出数组A，由你来计算fun(A)的结果。例如：A = {1, 4, 1}，fun(A) = [5/4] + [2/1] + [5/4] = 1 + 2 + 1 = 4。

Input

第1行：1个数N，表示数组A的长度(1 <= N <= 100000)。
第2 - N + 1行：每行1个数A[i]（1 <= A[i] <= 10^9)。

Output

输出fun(A)的计算结果。

Input示例

3
1 4 1

Output示例

上去直接怼暴力，上来就TLE了。什么时候我能分析分析算法的可行性呢？

网上思路（想不到啊想不到）：只有在少数情况下A[i]+A[j] 会大于等于 A[i]*A[j]，即(A[i]+A[j])/(A[i]*A[j])！=0；

1.A[i]=A[j]=2，取1；

2.A[i]、A[j]只有一个为 1 ，取1

3.A[i]=A[j]=1，取2

当2的个数为ans时，例如2 2 2 2，一共6个组合，即ans*(ans-1)/2；当1的个数为res时，一个1可以和n-1个数组合，每次+（n-1），最后sum=ans*(ans-1)/2+res*(n-1);

这里解释一下为什么1时为res*(n-1)；当第i个为1时，能和1组合的有n-1个，就是n-1个1，但是如果是第三种情况呢，其实我们已经计算进去了，例如1 2 1 2 1 2，当第一个1与第二个1组合时，应该取2，但是我们取了1，当我们计算第二个1时，我们又+（n-1），其实又加了回来。

#include<stdio.h>
#include<string.h>
using namespace std;
int main()
{
	int n,a;
	scanf("%d",&n);
	int ans=0;
	int k=0;
	for(int i=0;i<n;i++)
	{
		scanf("%d",&a);
		if(a==1)
			ans+=(n-1);
		else if(a==2)
			k++;
	}
	printf("%d\n",ans+(k-1)*k/2);
	return 0;
}