Submission #90077

#	Time	Username	Problem	Language	Result	Execution time	Memory
90077		Rashidov_Anis	은행 (IZhO14_bank)	C++14	71 / 100	1083 ms	756 KiB

bank

#include <bits/stdc++.h>
#define fr first
#define se second
#define sc scanf
#define pf printf
#define ld long double
#define ll long long
#define pb push_back
#define sq(n) n * n
#define r_0 return 0
#define pr pair < ll, ll >
#define prr pair < pair < ll, ll >, ll >
#define fin(s) freopen( s, "r", stdin )
#define fout(s) freopen( s, "w", stdout )
#define TIME ios_base::sync_with_stdio(0)
using namespace std;

const ll INF = 1e9;
const ll N = 2e3 + 3;
const ll mod = 998244353;
const ld eps = 1e-7;

ll n, m;
ll b[N], a[N];

bool dfs( ll x, ll sum, ll state, ll k, ll num ){
    if( x > n ){
        return 1;
    }
    if( sum == 0 ){
        return dfs( x + 1, a[x + 1], state, 1, num );
    }
    bool mx = 0;
    for( int i = k; i <= m; i ++ ){
        if( m - num < n - x )break;
        if( sum < b[i] || ( state >> i ) & 1 == 1 )continue;
        mx = max( dfs( x, sum - b[i], ( state | ( 1 << i ) ), i + 1, num + 1 ), mx );
        if( mx == 1 )break;
    }
    return mx;
}

main(){
    TIME;
    //fin( "bank.in" );
    //fout( "bank.out" );
    cin >> n >> m;
    for( int i = 1; i <= n; i ++ ){
        cin >> a[i];
    }
    for( int i = 1; i <= m; i ++ )cin >> b[i];
    ll x = dfs( 1, a[1], 0, 1, 0 );
    if( x == 0 ){ cout << "NO"; }
    else cout << "YES";
}

Compilation message (stderr)

bank.cpp: In function 'bool dfs(long long int, long long int, long long int, long long int, long long int)':
bank.cpp:36:46: warning: suggest parentheses around comparison in operand of '&' [-Wparentheses]
         if( sum < b[i] || ( state >> i ) & 1 == 1 )continue;
                                            ~~^~~~
bank.cpp: At global scope:
bank.cpp:43:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main(){
      ^

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