#include <bits/stdc++.h>
using namespace std;
double solve(int N, int M, int K, int H, std::vector<int> x, std::vector<int> y, std::vector<int> C, std::vector<int> arr){
double c[M];
for(int i = 0;i < M;i++) c[i] = C[i];
double dp[K+1];
fill(dp, dp+K+1, -1);
dp[0] = c[H-1];
if(arr[H-1] == 2){
dp[1] = c[H-1];
double mc = 0;
for(int i = 2;i <= K;i++) {
mc = mc*0.5 + c[H-2];
dp[i] = c[H-1]+mc;
}
}
for(int i = H-2;i >= 0;i--){
if(arr[i] == 2){
int m = min(c[i-1], c[i]);
for(int j = K;j >= 0;j--){
double ans = 1e20;
if(dp[j] != -1) ans = c[i]/(1<<j) + dp[j];
if(j && dp[j-1] != -1) ans = min(ans, c[i]/(1<<j-1) + dp[j-1]);
double mc = 0;
for(int k = 2;k <= j;k++){
mc = mc*0.5+m;
if(dp[j-k] != -1) ans = min(ans, c[i]/(1<<j-k) + dp[j-k] + mc/(1<<j-k));
}
if(ans < 1e20) dp[j] = ans;
else dp[j] = -1;
}
} else {
for(int j = 0;j <= K;j++) if(dp[j] != -1) dp[j] += c[i]/(1<<j);
if(!arr[i] || !i){
double ans = dp[0];
for(int j = 1;j <= K;j++) if(dp[j] != -1) ans = min(ans, dp[j]);
return ans;
}
}
}
return c[0];
}
int main(){
// cout << solve(6, 5, 2, 5, {0, 1, 2, 3, 4}, {1, 2, 3, 4, 5}, {1, 2, 4, 8, 16}, {1, 2, 0, 2, 2, 1}) << '\n';
cout << solve(3, 2, 3, 2, {0, 1}, {1, 2}, {1, 16}, {1, 2, 1});
}
Compilation message
cyberland.cpp: In function 'double solve(int, int, int, int, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
cyberland.cpp:23:61: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
23 | if(j && dp[j-1] != -1) ans = min(ans, c[i]/(1<<j-1) + dp[j-1]);
| ~^~
cyberland.cpp:27:60: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
27 | if(dp[j-k] != -1) ans = min(ans, c[i]/(1<<j-k) + dp[j-k] + mc/(1<<j-k));
| ~^~
cyberland.cpp:27:84: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
27 | if(dp[j-k] != -1) ans = min(ans, c[i]/(1<<j-k) + dp[j-k] + mc/(1<<j-k));
| ~^~
/usr/bin/ld: /tmp/ccUVk6I9.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/cceuqQR7.o:cyberland.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status