인프런 커뮤니티 질문&답변

안녕하세요 재욱님 ㅎㅎ

정말 잘 짜셨네요. ㅎㅎ

제가 주석 달아놨는데 참고 부탁드립니다.

#include <bits/stdc++.h>
using namespace std;


struct nut
{
    int a, b, c, d, e;
    
    nut(int a, int b, int c, int d, int e) : a(a), b(b), c(c), d(d), e(e) { }
    nut() { }
};

int n , m = 2100000000;

nut nu;
vector<nut> v;
vector<int> sel, ans;

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr); cout.tie(nullptr);
    
    int a, b, c, d, e;
    cin >> n;
    
    cin >> nu.a >> nu.b >> nu.c >> nu.d;
    for (int i = 0; i < n; i++)
    {
        cin >> a >> b >> c >> d >> e;
        v.push_back(nut(a, b, c, d, e));
    }
    
    for (int i = 1; i < (1 << n); i++)
    {
        //굿.
        nut nutri(0, 0, 0, 0, 0);
        sel.clear();
        for (int j = 0; j < n; j++)
        {
            if (i & (1 << j))sel.push_back(j);
        }
        
        for (int idx : sel){
            nutri.a += v[idx].a;
            nutri.b += v[idx].b;
            nutri.c += v[idx].c;
            nutri.d += v[idx].d;
            nutri.e += v[idx].e;
        } 

        if (nutri.a >= nu.a && nutri.b >= nu.b && nutri.c >= nu.c && nutri.d >= nu.d)
        {  
            if (m > nutri.e){    
                m = nutri.e;
                ans.clear();
                // 제가 j를 idx로 바꾼 이유는요. 이코드는 j여도 문제가 없어요.
                // 그치만 변수명 같은 경우 이렇게 j를 순회하면서 sel에다가 push하는 로직을 하고 여기서는 
                // sel의 요소를 넣을 때 j를 또 쓰잖아요. 이렇게 됬을 때 나중에 헷갈릴 수가 있어서 맞왜틀에 빠질 수 있거든요. 
                // 넉넉히 변수명 하나 더 씁시다.
                for (int idx : sel) ans.push_back(idx + 1);
            } // 굿. 
            else if (m == nutri.e){
                bool flag = false; 
                // 이부분!
                // ans 1 2 3 4
                // sel 1 2 3 인 경우가 반례입니다. 
                int size = min(ans.size(), sel.size());
                
                for (int j = 0; j < size; j++)
                {
                    if (ans[j] > sel[j] + 1){
                        flag = true;
                        break;
                    }
                }
                
                if (flag) {
                    ans.clear();
                    for (int idx : sel) ans.push_back(idx + 1);
                }
            }
        }
    
    }
    
    if (m == 2100000000) m = -1;
    //endl은 쓰지 말아주세요 - 교안 참고
    cout << m << endl; 
    cout << ba[0];
    if(m != -1){
        for (int i = 0; i < ans.size(); i++) cout << ans[i] << " ";
        cout << '\n'; 
    } 
}

 또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.