#include <bits/stdc++.h>
using namespace std;
using LL = long long int;
using uLL = unsigned long long int;
using uint = unsigned int;
using ld = long double;
const int N = 307;
const int MINUS = 2000000;
LL A[N];
int main(){
cin.tie(NULL);
ios_base::sync_with_stdio(0);
int n;
cin >> n;
for(int i = 1; i <= n; i++){
cin >> A[i];
}
vector <LL> ans((n*(n+1))/2+n), history((n*(n+1))/2+n+1);
LL rating = 0;
int pos = 1;
history[0] = 0;
for(int i = 1; i <= n; i++){
LL need = A[i];
//cout << "I " << i << '\n';
for(int j = 1; j <= i; j++){
LL inc = MINUS;
for(int k = 1; k <= j; k++){
inc = min(inc, history[pos-k]+A[k]-rating);
//cout << k << " " << history[pos-k] << " " << A[k] << " " << rating << '\n';
}
if(j == i){
if(need > inc){
cout << "NIE\n";
return 0;
}
inc = need;
}
ans[pos-1] = inc;
rating += inc;
need -= inc;
history[pos] = rating;
//cout << inc << '\n';
pos++;
}
//cout << '\n';
ans[pos-1] = -MINUS;
rating -= MINUS;
history[pos] = rating;
pos++;
}
cout << "TAK\n";
cout << ans.size() << '\n';
for(auto x: ans){
cout << x << " ";
}
cout << '\n';
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 | #include <bits/stdc++.h> using namespace std; using LL = long long int; using uLL = unsigned long long int; using uint = unsigned int; using ld = long double; const int N = 307; const int MINUS = 2000000; LL A[N]; int main(){ cin.tie(NULL); ios_base::sync_with_stdio(0); int n; cin >> n; for(int i = 1; i <= n; i++){ cin >> A[i]; } vector <LL> ans((n*(n+1))/2+n), history((n*(n+1))/2+n+1); LL rating = 0; int pos = 1; history[0] = 0; for(int i = 1; i <= n; i++){ LL need = A[i]; //cout << "I " << i << '\n'; for(int j = 1; j <= i; j++){ LL inc = MINUS; for(int k = 1; k <= j; k++){ inc = min(inc, history[pos-k]+A[k]-rating); //cout << k << " " << history[pos-k] << " " << A[k] << " " << rating << '\n'; } if(j == i){ if(need > inc){ cout << "NIE\n"; return 0; } inc = need; } ans[pos-1] = inc; rating += inc; need -= inc; history[pos] = rating; //cout << inc << '\n'; pos++; } //cout << '\n'; ans[pos-1] = -MINUS; rating -= MINUS; history[pos] = rating; pos++; } cout << "TAK\n"; cout << ans.size() << '\n'; for(auto x: ans){ cout << x << " "; } cout << '\n'; return 0; } |
English