Submission #1270162

#	Time	Username	Problem	Language	Result	Execution time	Memory
1270162		almaz	Detecting Molecules (IOI16_molecules)	C++20	31 / 100	585 ms	1284 KiB

molecules

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

// #define int long long
// #define endl '\n'
#define ff first
#define ss second
#define pb push_back
#define all(a) a.begin(), a.end()

// const int MOD = 1e9 + 7,INF = 1e18;

std::vector<int> find_subset(int l1, int u, std::vector<int> w) {
	ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    int n = w.size(), l = l1 , r = u;
    
    if(l == 1 && r == 1){
		while(true);
	}
	
	vector <int> a(n);
	
	int sum = 0;
	
	for(int i = 0;i < n;i++){
		a[i] = w[i];
		sum += a[i];
	}
	
	bitset<2000000> dp;
	
	dp[0] = 1;
	
	for(int i = 0;i < n;i++){
		bitset<2000000> f = dp << a[i];
		bitset<2000000> ndp = dp | f;
		
		dp = ndp;
	}
	int ans = -1;
	
	vector <int> b;
	
	for(int i = l;i <= r;i++){
		if(dp[i] == 1){
			ans = i;
			break;
		}
	}
	
	if(ans == -1){
		return b;
	}
	
	for(int i = 0;i < n;i++){
		if(ans - a[i] >= 0 && dp[ans - a[i]] == 1){
			ans -= a[i];
			b.pb(i);
		}
	}
	
	return b;
}

Compilation message (stderr)

molecules.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
molecules_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~

• Subtask 1: 9 / 9
• Subtask 2: 10 / 10
• Subtask 3: 12 / 12
• Subtask 4: 0 / 15
• Subtask 5: 0 / 23
• Subtask 6: 0 / 31