#include<bits/stdc++.h> #include "fish.h" using namespace std; #define ll long long long long max_weights(int n,int m,vector<int>x,vector<int>y,vector<int>w){ int count=0,X=-1,Y=-1; for(int i=0;i<m;i++){ if(x[i]%2==0){ count++; } X=max(X,x[i]); Y=max(Y,y[i]); } if(count==m){//subtask1 ll ans=0; for(int i=0;i<m;i++){ ans+=w[i]; } return ans; } if(X<=1){//subtask2 ll res=0; int dp[n+5][2]; for(int i=0;i<=n;i++){ dp[i][0]=0; dp[i][1]=0; } ll ans; for(int i=0;i<m;i++){ if(x[i]==1){ res+=w[i]; } dp[y[i]+1][x[i]]=w[i]; } if(n==2){ return max(dp[1][0]+dp[2][0],dp[1][1]+dp[2][1]); } ans=res; for(int i=1;i<=n;i++){ res+=dp[i][0]-dp[i][1]; ans=max(ans,res); } return ans; } if(Y==0){ int arr[n+5]; fill(arr,arr+n+3,0); for(int i=0;i<m;i++){ arr[x[i]+1]=w[i]; } ll dp[n+5][2][2];//current pos, current state,current-1 state dp[2][0][0]=0; dp[2][1][0]=arr[1]; dp[2][0][1]=arr[2]; dp[2][1][1]=0; for(int i=3;i<=n;i++){ dp[i][1][1]=max(dp[i-1][1][0],dp[i-1][1][1]); dp[i][1][0]=max(dp[i-1][0][1],dp[i-1][0][0]+arr[i-1]); dp[i][0][1]=max(dp[i-1][1][0],dp[i-1][1][1])+arr[i]; dp[i][0][0]=max(dp[i-1][0][1],dp[i-1][0][0]); } return max({dp[n][0][0],dp[n][0][1],dp[n][1][0],dp[n][1][1]}); } }

fish.cpp: In function 'long long int max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:64:1: warning: control reaches end of non-void function [-Wreturn-type]
   64 | }
      | ^