题目链接:传送门
题面
在[-30000,30000]范围里,给出一组整数集合S。找到满足的六元组的总数使得:
a ∗ b + c d − e = f \frac{a*b+c}{d}-e=f da∗b+c−e=f
并且保证元组 ( a , b , c , d , e , f ) : a , b , c , d , e , f ∈ S ; d ̸ = 0 (a,b,c,d,e,f):a,b,c,d,e,f \in S;d\not=0 (a,b,c,d,e,f):a,b,c,d,e,f∈S;d̸=0
输入格式
第一行包含整数N,代表集合的个数 接下来N行,每行包含一个整数,并且保证N个数互不相同
输出格式
输出可能组成满足题意的解的方案数 (a,b,c,d,e,f之间允许相等)
数据范围与提示
1⩽N⩽100
Step 1.爆搜?
枚举六个数的值,判断是否相等,时间复杂度
O
(
n
6
)
O(n^6)
O(n6)。
程序略。
Step 2.折半搜索
折半搜索内涵:把题目分成两部分解决,降低时间复杂度
这道题中,首先把等式变形为
a
∗
b
+
c
=
d
∗
(
e
+
f
)
a*b+c=d*(e+f)
a∗b+c=d∗(e+f)。
首先枚举出所有可能的a,b,c的值,算出
a
∗
b
+
c
a*b+c
a∗b+c的所有可能值,存在一个map里。
然后枚举出所有可能的d,e,f的值,求出此时的
d
∗
(
e
+
f
)
d*(e+f)
d∗(e+f),在map中找
a
∗
(
b
+
c
)
a*(b+c)
a∗(b+c)为这个值的次数,统计到ans中。
最后,ans即为最终答案。
时间复杂度:
O
(
n
3
l
o
g
3
n
)
O(n^3log^3n)
O(n3log3n) (枚举是
n
3
n^3
n3,map是
l
o
g
(
n
3
)
log(n^3)
log(n3))
Step 3.Coding!
#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<map>
#define re register int
using namespace std;
typedef long long ll;
int read()
{
re x=0,f=1;
char ch=getchar();
while(ch<'0' || ch>'9')
{
if(ch=='-') f=-1;
ch=getchar();
}
while(ch>='0' && ch<='9')
{
x=10*x+ch-'0';
ch=getchar();
}
return x*f;
}
const int Size=105;
const int INF=0x3f3f3f3f;
map<int,int> vis; //用map存所有的a*(b+c)的值,可以用哈希表代替
int n,a[Size];
int main()
{
n=read();
for(re i=1; i<=n; i++) a[i]=read();
for(re i=1; i<=n; i++)
{
for(re j=1; j<=n; j++)
{
for(re k=1; k<=n; k++)
{
int A=a[i],B=a[j],C=a[k];
vis[A*B+C]++;
}
}
}
int ans=0;
for(re i=1; i<=n; i++)
{
if(!a[i]) continue;
for(re j=1; j<=n; j++)
{
for(re k=1; k<=n; k++)
{
int D=a[i],E=a[j],F=a[k];
int sum=D*(E+F);
if(vis[sum])
{
ans+=vis[sum];
}
}
}
}
printf("%d",ans);
return 0;
}