Submission #467770

#	Time	Username	Problem	Language	Result	Execution time	Memory
467770		quocnghia32	Bank (IZhO14_bank)	C++14	0 / 100	1096 ms	24912 KiB

bank

#include <bits/stdc++.h>
#define TASK "Bank"
#define FAST ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define READFILE freopen(TASK".INP","r",stdin)
#define WRITEFILE freopen(TASK".OUT","w",stdout)
#define For(i,a,b) for(int i=a;i<=b;i++)
#define Rep(i,a,b) for(int i=a;i>=b;i--)
#define pb push_back
#define ENDL '\n'
#define debug(x) cout<<#x<<" = "<<x<<ENDL
#define fi first
#define se second
#define ever (;true;)
#define all(x) x.begin(),x.end()
#define sz(a) ((int)(a).size())
#define ms(a,x) memset(a,x,sizeof(a))
#define int long long

using namespace std;
typedef vector <int> vi;
typedef pair <int,int> ii;
typedef vector <ii> vpi;
typedef pair <int,ii> iii;
const int N = 20;
const int oo=0x3f;
const int mod=1e9+7;
const int dx[]={-1,0,1,0};
const int dy[]={0,-1,0,1};
int a[N], cost[N], trace[(1<<20)+5], b[N];
ii dp[(1<<20)+5];
void init(){
  FAST;

}
bool cmp(ii a, ii b){
    return (a.fi<b.fi || a.fi==b.fi && a.se<b.se);
}
void tracee(int mask){
    b[trace[mask]-1]=1;
    if (trace[mask]!=0) tracee(trace[mask]);
}
signed main()
{
  init();
  int t;
  cin >> t;
  while (t--){
    int n,m;
    ms(dp,0);
    ms(a,0);
    ms(cost,0);
    ms(trace,0);
    ms(b,0);
    cin >> n >> m;
    For(i,0,n-1) cin >> a[i];
    For(i,0,m-1) cin >> cost[i];
    int mm=1<<m;
    for (int mask=0; mask<mm; mask++){
        For(i,0,m-1){
            if ((mask>>i)&1){
                ii p=dp[mask^(1<<i)];
                if (p.se+cost[i]==a[p.fi]){
                    {
                        dp[mask]={p.fi+1,0};

                    }
                }
                else dp[mask]=max(dp[mask],{p.fi,p.se+cost[i]});
            }
        }
    }
    if (dp[mm-1].fi==n) {
        cout << "YES" << ENDL;
    } else cout << "NO" << ENDL;
  }
  return 0;
}

Compilation message (stderr)

bank.cpp: In function 'bool cmp(ii, ii)':
bank.cpp:36:37: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
   36 |     return (a.fi<b.fi || a.fi==b.fi && a.se<b.se);
      |                                     ^
bank.cpp: In function 'int main()':
bank.cpp:16:37: warning: 'void* memset(void*, int, size_t)' clearing an object of type 'ii' {aka 'struct std::pair<long long int, long long int>'} with no trivial copy-assignment; use assignment instead [-Wclass-memaccess]
   16 | #define ms(a,x) memset(a,x,sizeof(a))
      |                                     ^
bank.cpp:49:5: note: in expansion of macro 'ms'
   49 |     ms(dp,0);
      |     ^~
In file included from /usr/include/c++/10/bits/stl_algobase.h:64,
                 from /usr/include/c++/10/bits/char_traits.h:39,
                 from /usr/include/c++/10/ios:40,
                 from /usr/include/c++/10/istream:38,
                 from /usr/include/c++/10/sstream:38,
                 from /usr/include/c++/10/complex:45,
                 from /usr/include/c++/10/ccomplex:39,
                 from /usr/include/x86_64-linux-gnu/c++/10/bits/stdc++.h:54,
                 from bank.cpp:1:
/usr/include/c++/10/bits/stl_pair.h:211:12: note: 'ii' {aka 'struct std::pair<long long int, long long int>'} declared here
  211 |     struct pair
      |            ^~~~

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