제출 #656409

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
656409		asiat	Bank (IZhO14_bank)	C++14	52 / 100	1084 ms	73216 KiB

bank

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
const int N = 21;
int dp[N][(1<<N)];
main(){
	cin.tie(0)->sync_with_stdio(0);
	int n; cin >> n;
	int m; cin >> m;
	vector <int> v(n), g(m);
	vector <int> can[n+1];
	for (int i = 0; i < n; i++)cin >> v[i];
	for (int i = 0; i < m; i++)cin >> g[i];
	for (int i = 1; i < (1<<m); i++){
		int sum = 0;
		vector <int> ans;
		int x = i;
		while(x > 0){
			ans.push_back(x&1);
			x>>=1;
		}
		while(ans.size() < m)ans.push_back(0);
		reverse(ans.begin(), ans.end());
		for (int j = 0; j < m; j++)
			if(ans[j] == 1)sum+=g[j];
		for (int j = 0; j < n; j++)
			if(sum == v[j])can[j].push_back(i);
	}
	if(n > m){
		cout << "NO";
		return 0;
	}
	for (auto to : can[0]){
		dp[0][to] = 1;
	}
	vector <int> mask[n];
	for (int i = 0; i < (1<<m); i++){
		for (int j = 0; j < n; j++){
			for (auto to : can[j]){
				if(!(to&i))mask[j].push_back(i);
			}
		}
	}
	for (int i = 1; i < n; i++){
		for (int j = 0; j < mask[i].size(); j++){
			for (auto to : can[i]){
				if(!(mask[i][j] & to)){
					dp[i][mask[i][j]|to] |= dp[i-1][mask[i][j]];
				}
			}
		}
	}
	for (int i = 0; i < (1<<m); i++){
		int to = dp[n-1][i];
		if(to){
			cout << "YES" << endl;
			return 0;
		}
	}
	cout << "NO" << endl;
	return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

bank.cpp:7:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
    7 | main(){
      | ^~~~
bank.cpp: In function 'int main()':
bank.cpp:23:20: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   23 |   while(ans.size() < m)ans.push_back(0);
      |         ~~~~~~~~~~~^~~
bank.cpp:46:21: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   46 |   for (int j = 0; j < mask[i].size(); j++){
      |                   ~~^~~~~~~~~~~~~~~~

• Subtask 1: 0 / 19
• Subtask 2: 25 / 25
• Subtask 3: 27 / 27
• Subtask 4: 0 / 29