#include <bits/stdc++.h>
using namespace std;
const int maxk=404;
int tab[maxk], sum;
vector<int> gs;
int k = 400;
int n,r,g;
int main(){
scanf("%d", &n);
scanf("%d", &g);
gs.push_back(g);
k = min (n+4, k);
for(int i = 1; i <= k; i++){
tab[i] = g;
}
for(int i = 1 ; i < n; i++){
scanf("%d", &g);
gs.push_back(g);
sum = 0;
for(int p = 1; p <= i; p++){
sum += tab[p];
}
for(int q = 1; q <= k - i; q++){
sum -= tab[q-1];
sum += tab[q+i];
if(sum > g){
tab[q] -= (sum - g);
sum = g;
}
}
}
bool ans = true;
bool found;
for(int i = 0 ; i < n; i++){
sum = 0;
found = false;
for(int p = 1; p <= i; p++){
sum += tab[p];
}
for(int q = 1; q <= k - i; q++){
sum -= tab[q-1];
sum += tab[q+i];
if(sum == gs[i]){
found = true;
break;
}
}
if(found){
continue;
}
else{
ans = false;
break;
}
}
if(ans){
cout<<"TAK"<<endl;
printf("%d\n", k);
for(int i = 1; i<=k; i++){
printf("%d", tab[i]);
if(i<k) printf(" ");
}
printf("\n");
}
else{
cout<<"NIE"<<endl;
}
}
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 | #include <bits/stdc++.h> using namespace std; const int maxk=404; int tab[maxk], sum; vector<int> gs; int k = 400; int n,r,g; int main(){ scanf("%d", &n); scanf("%d", &g); gs.push_back(g); k = min (n+4, k); for(int i = 1; i <= k; i++){ tab[i] = g; } for(int i = 1 ; i < n; i++){ scanf("%d", &g); gs.push_back(g); sum = 0; for(int p = 1; p <= i; p++){ sum += tab[p]; } for(int q = 1; q <= k - i; q++){ sum -= tab[q-1]; sum += tab[q+i]; if(sum > g){ tab[q] -= (sum - g); sum = g; } } } bool ans = true; bool found; for(int i = 0 ; i < n; i++){ sum = 0; found = false; for(int p = 1; p <= i; p++){ sum += tab[p]; } for(int q = 1; q <= k - i; q++){ sum -= tab[q-1]; sum += tab[q+i]; if(sum == gs[i]){ found = true; break; } } if(found){ continue; } else{ ans = false; break; } } if(ans){ cout<<"TAK"<<endl; printf("%d\n", k); for(int i = 1; i<=k; i++){ printf("%d", tab[i]); if(i<k) printf(" "); } printf("\n"); } else{ cout<<"NIE"<<endl; } } |
English