이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
#define lg long long
#define fastio ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
const lg N = 5e3+5;
lg n, d, tim;
lg dp[N][N];
lg t[N], lst[N];
vector<lg> o[N];
lg C(lg i, lg j)
{
lg ans = o[j].size()-(lower_bound(o[j].begin(), o[j].end(), i)-o[j].begin());
return ans;
}
void compute(lg lev, lg l, lg r, lg optl, lg optr)
{
if(l > r) return;
lg mid = (l+r)/2;
array<lg, 2> best = {1000000000000, optl};
for(int i = optl; i <= min(optr, mid-1); i++)
{
best = min(best, {dp[lev+1][i]+C(i+1, mid), i});
}
dp[lev][mid] = best[0];
compute(lev, l, mid-1, optl, best[1]);
compute(lev, mid+1, r, best[1], optr);
return;
}
int main()
{
fastio;
cin >> n >> d >> tim;
for(int i = 1; i <= n; i++)
{
cin >> t[i];
}
for(int i = 1; i <= n; i++)
{
for(int j = i; j >= 1; j--)
{
if(t[j]+i-j <= tim)
{
lst[i] = j;
break;
}
}
o[i].push_back(lst[i]);
for(auto it : o[i-1]) o[i].push_back(it);
sort(o[i].begin(), o[i].end());
}
d = min(n, d);
t[0] = 1e18;
memset(dp, 0x3f, sizeof(dp));
dp[d][0] = 0;
for(int i = 1; i <= n; i++) dp[d][i] = C(1, i);
// for(int i = 1; i <= n; i++)
// {
// for(int j = 0; j < d; j++)
// {
// for(int k = i-1; k >= 0; k--)
// {
// dp[i][j] = min(dp[k][j+1]+C(k+1, i), dp[i][j]);
// }
// }
// }
for(int i = d-1; i >= 0; i--)
{
compute(i, 1, n, 1, n);
}
// for(int i = 1; i <= n; i++)
// {
// for(int j = 0; j <= d; j++)
// {
// cout << dp[j][i] << ' ';
// }
// cout << '\n';
// }
lg ans = n+1;
for(int j = 0; j <= d; j++)
{
ans = min(ans, dp[j][n]);
}
cout << ans << '\n';
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |