如果不是很确定枚举优化细节,可以先把最暴力的也就是最简单的写出来,然后再一步一步优化
思路进阶
①因为k取0-d,所以只要x+y+z在0-d内就有k可以满足,所以只要和在0-d内此时,x,y,z就有效,再嵌套3层循环分别对,x,y,z进行分别从0-a,0-b,0-c遍历,就可以得到排序不重复。
初步代码
#include <iostream>
using namespace std;
int main()
{
long long a,b,c,d;
long long int sum = 0;
while(cin >> a>> b>>c>>d)
{
sum = 0;
for(int i =0;i <= a;i++)
{
for(int j =0;j <= b;j++)
{
for(int k = 0;k <= c;k++)
{
if(i+j+k<=d)
{
sum++;
}
}
}
}
cout << sum <<endl;
}
system("pause");
return 0;
}
②题目数据0-10^6,所以明显要优化到只有1个for循环。
发现确定x后,用z-x的值和y+z的值进行比较,针对不同比较结果有不同的满足z-x>=y+z的序列的个数的公式。
一下代码中j=z-x;
#include <iostream>
using namespace std;
int main()
{
long long int a,b,c,d,j;
long long int sum = 0;
while(cin >> a>> b>>c>>d)
{
sum = 0;
for(long long int i =0;i <= a && i<=d;i++)
{
j = d-i;
if(j == 0)
sum++;
else
{
if(j>=b+c)
sum = sum + (b+1)*(c+1);
else
{
if(c>=j && b>=j)
sum+=(j+2)*(j+1)/2;
else if(c>=j && b<j)
sum+=(2*j+2-b)*(b+1)/2;
else if(b>=j && c < j)
sum+=(2*j+2-c)*(c+1)/2;
else
{
for(int key = 0;key<=b;key++)
{
if(j>=key+c)
sum+=(c+1);
else
{
sum+=(j-key+1);
}
}
}
/*for(long long int k =0;k <= b && k<=j;k++)
{
if(k+c<=j)
sum+=(c+1);
else
{
sum+=(j-k+1);
}
}*/
}
}
}
cout << sum <<endl;
}
//system("pause");
return 0;
}