一、题目链接
二、题目分析
(一)算法标签
枚举 组合数学
(二)解题思路
一、先枚举1、2位,并记录它们的和的方案数,
二、再枚举3、4位,并记录它们的和的方案数,
最后枚举一、二,判断第五个数是否合法,并根据乘法原理更新答案
三、AC代码
解法一:(运行时间0.047s)
#include <iostream>
using namespace std;
typedef long long LL;
const int N = 2021;
// cnt1[i]表示第1、2个数的和为i的方案个数
// cnt2[i]表示第3、4个数的和为i的方案个数
int cnt1[2030], cnt2[2030];
LL ans = 0;
int main()
{
for (int i = 1; i < N; ++ i)
{
for (int j = 1; j < N ; ++ j)
{
if (i + j >= N) break;
cnt1[i + j] ++ ;
}
}
for (int i = 1; i < N; i ++ )
for (int j = 1; j < N; j ++ )
{
if (i + j >= N) break;
cnt2[i + j] ++ ;
}
for (int i = 2; i <= N - 3; i ++ )
for (int j = 2; j <= N - 3; j ++ )
{
if (N - i - j <= 0 || i + j < 4) break;
ans += (LL)cnt1[i]*cnt2[j];
}
printf("%lld", ans); //691677274345
return 0;
}
解法二(DFS):(严重TLE,仅提供一种思路)
#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
// num个数相加的和为N
const int N = 5, num = 3;
typedef long long LL;
vector<int> way;
LL ans = 0;
void dfs(int cnt, LL sum)
{
if (cnt > num || sum > N) return;
if (sum == N && cnt == num)
{
ans ++ ;
for (auto it : way)
printf("%d ", it);
puts("");
return;
}
for (int i = 1; i <= N; i ++ )
{
sum += i;
way.push_back(i);
dfs(cnt + 1, sum);
way.pop_back();
sum -= i;
}
}
int main()
{
dfs(0, 0);
printf("%d", ans);
return 0;
}
题目示例运行结果: