Submission #499151

#	Time	Username	Problem	Language	Result	Execution time	Memory
499151		beksultan04	Bank (IZhO14_bank)	C++14	71 / 100	1092 ms	51488 KiB

bank

#include <bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
//#define int long long
#define pii pair<int,int>
#define ret return
#define fr first
#define sc second
#define OK puts("OK");
#define NO puts("NO");
#define YES puts("YES");
#define all(s) s.begin(),s.end()
#define allr(s) s.rbegin(),s.rend()
#define nosol puts("-1");
#define pb push_back
#define endi puts("");
#define ordered_set tree <int, null_type,less<int>, rb_tree_tag,tree_order_statistics_node_update>
const int N = 2e6+12,INF = 1e18+7;

vector <int> g[N];
int a[N],b[N],dp[N];
main(){
	int n,i,j,m,mask;
	cin>>n>>m;
	for (i=0;i<n;++i)scanf("%d",&a[i]);
	for (j=0;j<m;++j)scanf("%d",&b[j]);
		
	for (i=0;i<n;++i){
		for (mask = 0;mask<(1<<m);++mask){
			int sum = 0;
			for (j=0;j<m;++j){
				if (mask&(1<<j))
					sum += b[j];
			}
			if (sum == a[i])g[i].pb(mask);
		}
	}
	dp[0] = 1;
	for (i=0;i<n;++i){
		for (mask = (1<<m);mask >= 0; --mask){
			if (dp[mask] == i+1){
				for (auto ms: g[i]){
					if (ms&mask)continue;
					int x = mask|ms;
					
					dp[x] = max(dp[x],dp[mask]+1);
					if (dp[x]> n){
						YES
						ret 0;
					}
					
				}
				
			}
		}
	}
	cout <<"NO";
}

Compilation message (stderr)

bank.cpp:20:32: warning: overflow in conversion from 'double' to 'int' changes value from '1.0e+18' to '2147483647' [-Woverflow]
   20 | const int N = 2e6+12,INF = 1e18+7;
      |                            ~~~~^~
bank.cpp:24:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   24 | main(){
      | ^~~~
bank.cpp: In function 'int main()':
bank.cpp:27:24: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   27 |  for (i=0;i<n;++i)scanf("%d",&a[i]);
      |                   ~~~~~^~~~~~~~~~~~
bank.cpp:28:24: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   28 |  for (j=0;j<m;++j)scanf("%d",&b[j]);
      |                   ~~~~~^~~~~~~~~~~~

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