day 15: 78-subsets

This commit is contained in:
Kaushik Narayan R 2023-10-14 10:18:35 -07:00
parent 7375af90a8
commit 3de8ed3745
3 changed files with 41 additions and 0 deletions

15
78-subsets/driver.cpp Normal file
View File

@ -0,0 +1,15 @@
#include "soln.hpp"
int main() {
vector<int> nums{1,2,3};
Solution soln;
vector<vector<int>> answer = soln.subsets(nums);
for(int i = 0; i < answer.size(); i++) {
cout << "Subset " << (i+1) << ": ";
for(int j = 0; j < answer[i].size(); j++) {
cout << answer[i][j] << " ";
}
cout << endl;
}
return 0;
}

19
78-subsets/soln.cpp Normal file
View File

@ -0,0 +1,19 @@
#include "soln.hpp"
vector<vector<int>> Solution::subsets(vector<int> &nums)
{
int num_combos = 1 << nums.size();
vector<vector<int>> subs(num_combos);
for (int i = 0; i < num_combos; i++)
{
for (int j = 0; j < nums.size(); j++)
{
// check if this index exists in this combo
if ((i >> j) & 1)
{
subs[i].push_back(nums[j]);
}
}
}
return subs;
}

7
78-subsets/soln.hpp Normal file
View File

@ -0,0 +1,7 @@
#include <bits/stdc++.h>
using namespace std;
class Solution
{
public:
vector<vector<int>> subsets(vector<int> &nums);
};