装箱问题c++

代码

//解法1：贪心
#include <bits/stdc++.h>
using namespace std;
int status[8][4];//status[i][j]:包裹中已有1个大小为i*i的产品，此时可以最多放入多少个j*j的产品 
void initStatus()
{//status[0]:包裹中没有产品时，各种产品最多可以放多少个 
    status[2][2] = 8;
    status[3][2] = 5, status[3][3] = 3;
    status[4][2] = 5;
    for(int i = 1; i <= 6; ++i)
        status[i][1] = 36 - i*i;
}
int pro[8];//pro[i]:i*i产品的数量
int main() 
{
    initStatus(); 
    while(true)
    {
        int sum = 0, bag = 0, rem[8];//sum:总产品数量  bag:包裹数 rem[i]:当前剩余空间能放几个i*i 
        for(int i = 1; i <= 6; ++i)
        {
            cin >> pro[i];
            sum += pro[i];
        }
        if(sum == 0)//如果6个数都是0，那么加和为0，要跳出循环 
            break;
        for(int i = 6; i >= 1; --i)
        {
            while(pro[i] > 0)
            {
                bag++;//用一个新包裹放i*i
                pro[i]--;
                for(int j = 1; j <= 3; ++j)//获取当前剩余空间情况 
                    rem[j] = status[i][j];
                int j = 3;
                while(j >= 1)
                {
                    if(rem[j] > 0 && pro[j] > 0)//如果可以放j*j的产品 
                    {
                        pro[j]--;//放一个产品 
                        if(j == 3)
                        {
                            rem[3]--;
                            rem[2] -= 2;
                            rem[1] -= 9;
                        }
                        else if(j == 2)
                        {
                            rem[2]--;
                            rem[1] -= 4;
                        }
                        else//j == 1
                            rem[1]--;
                    }
                    else
                        --j;
                }
            }   
        }
        cout << bag << endl;
    }  
    return 0;
}