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

안창완님의 프로필 이미지
안창완

작성한 질문수

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

4-A

4-A 질문 있습니다.

해결된 질문

작성

·

15

·

수정됨

0

- 학습 관련 질문을 남겨주세요. 상세히 작성하면 더 좋아요!
- 먼저 유사한 질문이 있었는지 검색해보세요.
- 서로 예의를 지키며 존중하는 문화를 만들어가요.
- 잠깐! 인프런 서비스 운영 관련 문의는 1:1 문의하기를 이용해주세요.

안녕하세요 선생님 이 문제를 재귀를 이용해서 풀어봤는데 인덱스를 포함하는 경우에서 temp_v 배열에 인덱스를 포함해서 보내는 경우는 어떻게 하는건가요??

(요 부분)

// 현재 인덱스를 포함하지 않는 경우
    go(idx + 1, p, f, s, v, sum, temp_v);

    // 현재 인덱스를 포함하는 경우
    go(idx + 1, 
       p + a[idx].p, 
       f + a[idx].f, 
       s + a[idx].s, 
       v + a[idx].v, 
       sum + a[idx].c, 
       temp_v);
#include <bits/stdc++.h>
using namespace std;

const int INF = 987654321;
int n, mp, mf, ms, mv;
int ret = INF;

map<int, vector<vector<int>>> ret_v; // 비용별 조합 저장

struct A {
    int p, f, s, v, c;
} a[16];

void go(int idx, int p, int f, int s, int v, int sum, vector<int> temp_v) {
    // 모든 재료를 확인한 경우
    if (idx == n) {
        // 최소 조건 만족하는지 확인
        if (p >= mp && f >= mf && s >= ms && v >= mv) {
            if (ret > sum) {
                ret = sum; // 최소 비용 갱신
                ret_v[ret].clear(); // 이전 조합 제거
                ret_v[ret].push_back(temp_v); // 새로운 조합 저장
            } else if (ret == sum) {
                ret_v[ret].push_back(temp_v); // 같은 비용의 다른 조합 저장
            }
        }
        return;
    }

    // 현재 인덱스를 포함하지 않는 경우
    go(idx + 1, p, f, s, v, sum, temp_v);

    // 현재 인덱스를 포함하는 경우
    go(idx + 1, 
       p + a[idx].p, 
       f + a[idx].f, 
       s + a[idx].s, 
       v + a[idx].v, 
       sum + a[idx].c, 
       temp_v);
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    // 입력 받기
    cin >> n >> mp >> mf >> ms >> mv;
    for (int i = 0; i < n; i++) {
        cin >> a[i].p >> a[i].f >> a[i].s >> a[i].v >> a[i].c;
    }

    // 탐색 시작
    go(0, 0, 0, 0, 0, 0, {});

    // 결과 처리
    if (ret == INF) {
        cout << -1 << '\n'; // 조건 만족하지 못하는 경우
    } else {
        sort(ret_v[ret].begin(), ret_v[ret].end()); // 사전순 정렬
        cout << ret << '\n'; // 최소 비용 출력
        for (int idx : ret_v[ret][0]) { // 첫 번째 조합 출력
            cout << idx << " "; 
        }
        cout << '\n';
    }
    return 0;
}

답변 1

0

큰돌님의 프로필 이미지
큰돌
지식공유자

안녕하세요 창완님 ㅎㅎ

저거 저 담는 경우의 수 함수 위 아래로

push 하구

pop_back 써서 원복시키시면 되요!!

 

또는 배열 안쓰구 비트마스킹 써서 해두 되구요ㅎㅎ

 

하시고 또 질문있으시면 질문주세요 ㅎㅎ

감사합니다

안창완님의 프로필 이미지
안창완

작성한 질문수

질문하기