#include <bits/stdc++.h>
using namespace std;
int n;
vector<pair<int, int>> adj[310];
vector<int> waga(310, 1e9);
vector<int> tab(310, 1e9);
int main(){
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
cin >> n;
waga[0] = 0;
tab[0] = 0;
for (int i=1; i<=n; ++i) {
int x;
cin >> x;
tab[i] = x;
for (int j=0; j<=n-i; ++j) {
adj[j].push_back({i+j, x});
}
}
/*
for (int i=0; i<=n; ++i){
cout << i << ": ";
for (auto u: adj[i]) {
cout << "(" << u.first << ", " << u.second << ") ";
} cout << endl;
} */
for (int i=0; i<n; ++i) {
for (auto u: adj[i]) {
waga[u.first] = min(waga[u.first], waga[i] + u.second);
}
}
/*
for (int i=1; i<=n; ++i){
cout << waga[i] << ' ';
}
cout << endl;
for (int i=1; i<=n; ++i){
cout << waga[i] - waga[i-1] << ' ';
} cout << endl;
*/
bool exist = true;
for (int i=1; i<=n; ++i) {
if (waga[i] != tab[i]) {
exist = false;
}
}
if (exist) {
cout << "TAK\n";
cout << n << endl;
for (int i=1; i<=n; ++i)
cout << tab[i] - tab[i-1] << ' ';
} else cout << "NIE";
}
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 | #include <bits/stdc++.h> using namespace std; int n; vector<pair<int, int>> adj[310]; vector<int> waga(310, 1e9); vector<int> tab(310, 1e9); int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); cin >> n; waga[0] = 0; tab[0] = 0; for (int i=1; i<=n; ++i) { int x; cin >> x; tab[i] = x; for (int j=0; j<=n-i; ++j) { adj[j].push_back({i+j, x}); } } /* for (int i=0; i<=n; ++i){ cout << i << ": "; for (auto u: adj[i]) { cout << "(" << u.first << ", " << u.second << ") "; } cout << endl; } */ for (int i=0; i<n; ++i) { for (auto u: adj[i]) { waga[u.first] = min(waga[u.first], waga[i] + u.second); } } /* for (int i=1; i<=n; ++i){ cout << waga[i] << ' '; } cout << endl; for (int i=1; i<=n; ++i){ cout << waga[i] - waga[i-1] << ' '; } cout << endl; */ bool exist = true; for (int i=1; i<=n; ++i) { if (waga[i] != tab[i]) { exist = false; } } if (exist) { cout << "TAK\n"; cout << n << endl; for (int i=1; i<=n; ++i) cout << tab[i] - tab[i-1] << ' '; } else cout << "NIE"; } |
English