题目
D、乐趣
每次测试时间限制:2秒
每个测试的内存限制:256 MB
给定两个整数 n 和 x,求满足 ab + ac + bc < n 且 a + b + c < x 的正整数三元组 (a, b, c) 的数量。请注意顺序问题(例如,(1, 1, 2) 和 (1, 2, 1) 被
视为不同),并且 a、b、c 必须严格大于 0。
输入
第一行包含一个整数 t (1 ≤ t ≤ 10^4) - 测试用例的数量。
每个测试用例包含两个整数 n 和 x (1 ≤ n, x ≤ 10^6)。
保证所有测试用例的n之和不超过106,并且所有测试用例的x之和不超过106。
输出
输出一个整数,即满足 ab + ac + bc < n 且 a + b + c < x 的正整数三元组 (a, b, c) 的数量。
题目分析
题目让求得的是满足条件的三元组的数目,而三元组的限制条件是一个跟三个元素均有关系的两个不等式。
要点
- 如果某些没带脑子的小朋友一眼之下,感觉光锁定a,b这两个元素就需要n^2的时间复杂度,一眼卡死。实际上,由于限制条件,在遍历了所有a的可能也就是O(n)的时间复杂度以后,在两个限制条件下锁定b最多只需要 ∑ i = 1 n n ∗ n i \sum_{i=1}^{n}{\frac{n*n}{i}} i=1∑nin∗n的时间复杂度
2.还有就是意识到如果锁定了a与b以后,可以通过不等式用a,b,n,x来表示c的范围,从而在两个范围之间选一个 a b + a c + b c < = x 可以转化为 a b + c ( a + b ) < = n 进一步可以表示为 c < = n − a b a + b a + b + c < = x 就更简单了,转化为 c < = x − a − b ab+ac+bc<=x可以转化为ab+c(a+b)<=n \newline 进一步可以表示为c<=\frac{n-ab}{a+b}\newline a+b+c<=x就更简单了,转化为c<=x-a-b ab+ac+bc<=x可以转化为ab+c(a+b)<=n进一步可以表示为c<=a+bn−aba+b+c<=x就更简单了,转化为c<=x−a−b
到时候确定了a和b直接可以从这两个范围中选取一个较小的然后加上去就是c的个数
解答
完整代码
key part
inline void solve() {
int n,x;cin>>n>>x;
ll ans=0;
rep(a,1,n-2){
for(int b=1;b<n/a&&b<x-a;b++){
ans += min((n-a*b)/(a+b),x-a-b);
}
}
cout<<ans<<endl;
}
详细说明该双重循环下的时间复杂度
int n,x;cin>>n>>x;
ll ans=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=n/i;j++){
ans++;
}
}
cout<<ans<<endl;
该代码外层循环n次内层循环n/i次,整体上时间复杂度换一句话说就是n乘上了个调和级数,而调和级数时间复杂度:
因此本题时间复杂度相当于O(10n)也就是1e6,2s足够了。
other parts
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define For for (ll i = 1; i <= n; i++)
#define rFor for (ll i = n; i > 0; i--)
#define rep(i, sta, end) for (ll i = sta; i <= end; i++)
#define rrep(i, end, sta) for (ll i = end; i >= sta; i--)
#define ALL(x) for (auto item : x)
……
int main() {
ios::sync_with_stdio(0);
cin.tie(0), cout.tie(0);
sieve(1e5);
ll num = 1;
cin >> num;
while (num--) {
//cout << "Main function execute properly before solve function " << endl;
solve();
// cout << "Main function execute properly after solve function" << endl;
}
return 0;
}