#include <iostream>
#include <vector>
using namespace std;
const int MAXN = 100005;
int T[MAXN];
int S[MAXN];
int cnt[MAXN];
int da_sie[MAXN]; // 0 --> nie da sie; 1 --> da sie; -1 --> nieokreslone
int n;
int czy_k(int k)
{
for (int i = 0; i <= n; i++)
{
S[i] = 0;
cnt[i] = 0;
}
for (int i = 0; i < n; i++)
{
if (i > 0)
S[i] += S[i - 1];
cnt[i] += S[i];
if (cnt[i] > T[i])
return 0;
else if (cnt[i] == T[i])
continue;
else
{
if (i + k > n)
return 0;
S[i] += T[i] - cnt[i];
S[i + k] -= T[i] - cnt[i];
cnt[i] = T[i];
}
}
return 1;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n;
for (int i = 0; i <= n; i++)
da_sie[i] = -1;
for (int i = 0; i < n; i++)
cin >> T[i];
for (int i = n; i >= 1; i--)
{
int t = i;
vector<int> czynniki;
for (int k = 2; k * k <= t; k++)
{
if (t % k == 0)
czynniki.push_back(k);
while (t % k == 0)
t /= k;
}
if (t > 1)
czynniki.push_back(t);
// for (int j = 0; j < czynniki.size(); j++)
// cerr << czynniki[j] << " ";
// cerr << endl;
bool ok = true;
for (int j = 0; j < czynniki.size(); j++)
{
if (da_sie[czynniki[j]] == -1)
da_sie[czynniki[j]] = czy_k(czynniki[j]);
if (da_sie[czynniki[j]] == 0)
{
ok = false;
break;
}
}
if (!ok)
continue;
if (da_sie[i] == -1)
da_sie[i] = czy_k(i);
if (da_sie[i] == 1)
{
cout << i << '\n';
return 0;
}
// cerr << 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 75 76 77 78 79 80 81 82 83 84 | #include <iostream> #include <vector> using namespace std; const int MAXN = 100005; int T[MAXN]; int S[MAXN]; int cnt[MAXN]; int da_sie[MAXN]; // 0 --> nie da sie; 1 --> da sie; -1 --> nieokreslone int n; int czy_k(int k) { for (int i = 0; i <= n; i++) { S[i] = 0; cnt[i] = 0; } for (int i = 0; i < n; i++) { if (i > 0) S[i] += S[i - 1]; cnt[i] += S[i]; if (cnt[i] > T[i]) return 0; else if (cnt[i] == T[i]) continue; else { if (i + k > n) return 0; S[i] += T[i] - cnt[i]; S[i + k] -= T[i] - cnt[i]; cnt[i] = T[i]; } } return 1; } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cin >> n; for (int i = 0; i <= n; i++) da_sie[i] = -1; for (int i = 0; i < n; i++) cin >> T[i]; for (int i = n; i >= 1; i--) { int t = i; vector<int> czynniki; for (int k = 2; k * k <= t; k++) { if (t % k == 0) czynniki.push_back(k); while (t % k == 0) t /= k; } if (t > 1) czynniki.push_back(t); // for (int j = 0; j < czynniki.size(); j++) // cerr << czynniki[j] << " "; // cerr << endl; bool ok = true; for (int j = 0; j < czynniki.size(); j++) { if (da_sie[czynniki[j]] == -1) da_sie[czynniki[j]] = czy_k(czynniki[j]); if (da_sie[czynniki[j]] == 0) { ok = false; break; } } if (!ok) continue; if (da_sie[i] == -1) da_sie[i] = czy_k(i); if (da_sie[i] == 1) { cout << i << '\n'; return 0; } // cerr << endl; } } |
English