Easy Equation

如果不是很确定枚举优化细节,可以先把最暴力的也就是最简单的写出来,然后再一步一步优化

Easy Equation

思路进阶

①因为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;
}

猜你喜欢

转载自blog.csdn.net/RunningBeef/article/details/109409235