#include <bits/stdc++.h> using namespace std; typedef long long LL; typedef long double LD; typedef unsigned long long ULL; typedef vector<int> VI; typedef vector<LL> VLL; typedef set<int> SI; typedef pair<int, int> PII; typedef pair<LL, LL> PLL; typedef vector<PII> VPII; typedef vector<PLL> VPLL; const int INF = 1000000001; const LD EPS = 1e-9; const int MOD = 1000000007; const LL LLINF = 1000000000000000001; #define FOR(i, b, e) for(int i = b; i <= e; i++) #define FORD(i, b, e) for(int i = b; i >= e; i--) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define DPRINT(x) cout << #x << ": " << x << endl #define BOOST ios_base::sync_with_stdio(false); cin.tie(nullptr) #define MP make_pair #define PB push_back #define ST first #define ND second #define GGL(x) "Case #" << x << ": " // ゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴ void no() { cout << "NIE\n"; exit(0); } int main() { BOOST; int n; cin >> n; VLL s(n+1); VLL ans(1); cin >> s[1]; ans.PB(s[1]); FOR(i, 2, n) { cin >> s[i]; ans.PB(s[i] - s[i-1]); } FOR(l, 1, n) { LL sum = 0; LL ma = -LLINF; FOR(i, 1, l-1) sum += ans[i]; FOR(i, l, n) { sum += ans[i]; ma = max(ma, sum); sum -= ans[i + 1 - l]; } if (ma != s[l]) no(); } cout << "TAK\n" << n << "\n"; FOR(i, 1, n) cout << ans[i] << ' '; cout << '\n'; }
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 76 77 78 79 80 81 | #include <bits/stdc++.h> using namespace std; typedef long long LL; typedef long double LD; typedef unsigned long long ULL; typedef vector<int> VI; typedef vector<LL> VLL; typedef set<int> SI; typedef pair<int, int> PII; typedef pair<LL, LL> PLL; typedef vector<PII> VPII; typedef vector<PLL> VPLL; const int INF = 1000000001; const LD EPS = 1e-9; const int MOD = 1000000007; const LL LLINF = 1000000000000000001; #define FOR(i, b, e) for(int i = b; i <= e; i++) #define FORD(i, b, e) for(int i = b; i >= e; i--) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define DPRINT(x) cout << #x << ": " << x << endl #define BOOST ios_base::sync_with_stdio(false); cin.tie(nullptr) #define MP make_pair #define PB push_back #define ST first #define ND second #define GGL(x) "Case #" << x << ": " // ゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴゴ void no() { cout << "NIE\n"; exit(0); } int main() { BOOST; int n; cin >> n; VLL s(n+1); VLL ans(1); cin >> s[1]; ans.PB(s[1]); FOR(i, 2, n) { cin >> s[i]; ans.PB(s[i] - s[i-1]); } FOR(l, 1, n) { LL sum = 0; LL ma = -LLINF; FOR(i, 1, l-1) sum += ans[i]; FOR(i, l, n) { sum += ans[i]; ma = max(ma, sum); sum -= ans[i + 1 - l]; } if (ma != s[l]) no(); } cout << "TAK\n" << n << "\n"; FOR(i, 1, n) cout << ans[i] << ' '; cout << '\n'; } |