#include <iostream> const int MAX = 303; int S[MAX]; int V[MAX]; int K[MAX]; int n; int INF = MAX * 1000000; bool check(int s) { int sum = 0; for (int i=0;i<=s;++i) sum += V[i]; for (int i=s;i<n;++i) { if (sum > S[s]) return false; sum += V[i]; sum -= V[i]; } return sum <= S[s]; } bool check2(int s) { int sum = 0; for (int i=0;i<=s;++i) sum += V[i]; for (int i=s;i<n;++i) { if (sum == S[s]) return true; sum += V[i]; sum -= V[i]; } return sum == S[s]; } int main() { std::ios_base::sync_with_stdio(0); std::cin >> n; for (int i=0;i<n;++i) std::cin >> S[i]; for (int i=0;i<n;++i) { for (int j=0;j<i;++j) { K[j] -= V[i-j-1]; } V[i] = INF; for (int j=0;j<=i;++j) { V[i] = std::min(V[i], S[j]-K[j]); } // for (int j=0;j<=i;++j) // { // std::clog << ":: $$ :: " << K[j] << std::endl; // } for (int j=0;j<=n;++j) { K[j] += V[i]; } // std::clog << ":: " << V[i] << " / " << S[i] << std::endl; // for (int j=0;j<=i;++j) // { // std::clog << ":: :: " << K[j] << std::endl; // } // std::clog << std::endl; } // for (int i=0;i<n;++i) // std::clog << V[i] << " "; // std::clog << std::endl; bool result = true; for (int i=0;i<n;++i) result &= check(i) & check2(i); if (result) { std::cout << "TAK\n" << n << std::endl; for (int i=0;i<n;++i) std::cout << V[i] << " "; std::cout << std::endl; } else { std::cout << "NIE" << std::endl; } 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 64 65 66 67 68 69 70 71 72 73 74 75 | #include <iostream> const int MAX = 303; int S[MAX]; int V[MAX]; int K[MAX]; int n; int INF = MAX * 1000000; bool check(int s) { int sum = 0; for (int i=0;i<=s;++i) sum += V[i]; for (int i=s;i<n;++i) { if (sum > S[s]) return false; sum += V[i]; sum -= V[i]; } return sum <= S[s]; } bool check2(int s) { int sum = 0; for (int i=0;i<=s;++i) sum += V[i]; for (int i=s;i<n;++i) { if (sum == S[s]) return true; sum += V[i]; sum -= V[i]; } return sum == S[s]; } int main() { std::ios_base::sync_with_stdio(0); std::cin >> n; for (int i=0;i<n;++i) std::cin >> S[i]; for (int i=0;i<n;++i) { for (int j=0;j<i;++j) { K[j] -= V[i-j-1]; } V[i] = INF; for (int j=0;j<=i;++j) { V[i] = std::min(V[i], S[j]-K[j]); } // for (int j=0;j<=i;++j) // { // std::clog << ":: $$ :: " << K[j] << std::endl; // } for (int j=0;j<=n;++j) { K[j] += V[i]; } // std::clog << ":: " << V[i] << " / " << S[i] << std::endl; // for (int j=0;j<=i;++j) // { // std::clog << ":: :: " << K[j] << std::endl; // } // std::clog << std::endl; } // for (int i=0;i<n;++i) // std::clog << V[i] << " "; // std::clog << std::endl; bool result = true; for (int i=0;i<n;++i) result &= check(i) & check2(i); if (result) { std::cout << "TAK\n" << n << std::endl; for (int i=0;i<n;++i) std::cout << V[i] << " "; std::cout << std::endl; } else { std::cout << "NIE" << std::endl; } return 0; } |