不等式的整数解（C++）

代码：

//
// Created by LWJ on 2024-11-21 9:11.
//

/*
 * 用回溯法求解下列不等式的所有的整数解。给出伪码和解。
 * 3*x1+4*x2+2*x3<= 12
 * x1, x2 ,x3 为非负整数
 */

#include <iostream>
using namespace std;

void inequation(int p[], int n, int b)
{
    int i=0, sum=0; // i表示第i个分量，sum前i个分量的多项式之和
    int k[n], count=0; // k[i]存储第i个分量的解，count表示解的个数
    while(1) {
        for (int x=0; x<b; x++) {
            if (sum+p[i]*x > b) break; // 退出for循环
            if (i<n-1) { // 进入下一层
                sum += p[i]*x;
                k[i]=x;
                i+=1;
                x=-1;
            } else { // 输出一个解：k[0], k[1], ..., k[n-1]
                k[i] = x;
                // cout << "解：";
                cout << "(";
                for (int j=0; j<n; j++) cout << k[j] << " ";
                cout << ")";
                count+=1;
                if(count%10==0) cout << endl;
            }
        }
        // 求下一个解
        
        while(i>0&&sum+p[i-1]>b){ // 上一层也到了极限，在向上一层回溯
            i-=1;
            sum -= p[i]*k[i]; // 向量上移一位，总和也要减去一位
        }
        if(i==0) break; // 结束算法，退出while循环
        k[i-1]+=1; // 上一个分量的x值+1
        sum+=p[i-1]; // 由于上一个分量的x值+1，所以sum也要加上p[i-1]
    }
    cout << endl;
    cout << "count=" << count<< endl;
}

int main()
{ // 3*x1+4*x2+2*x3<=12
    int p[]={3,4,2};
    int b=12;
    int n=sizeof(p)/sizeof(p[0]);
    inequation(p,n,b);
    return 0;
}

运行截图：