제출 #626604

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
626604		whynotshadow	Catfish Farm (IOI22_fish)	C++17	12 / 100	110 ms	7280 KiB

fish

#include "fish.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const ll N=3e5+9;
const ll INF=1e9;
ll used[N];
ll dp[N][3];
long long max_weights(int N, int M, std::vector<int> X, std::vector<int> Y, std::vector<int> W) {
  ll ans=0,cnt=0,cntt=0,cnt2=0,sum1=0,sum2=0;
  for(auto to:W){
    ans+=to;
  }
  for(auto to:X){
    if(to&1) cnt++;
    if(to<=1) cntt++;
  }
  for(auto to:Y){
    if(to==0) cnt2++;
  }
  if(cnt==0) return ans;
  else if(cnt2==M){
    for(ll i=0; i<=N; i++){
        used[i]=0;
        dp[i][0]=dp[i][1]=0;
    }
    for(ll i=0;i<M;i++){
        used[X[i]+1]=W[i];
    }
//    dp[-1][0]=dp[-1][1]=0;
    /// 0 - used, 1 - free
    for(ll i=1;i<=N;i++){
        if(i==1){
            dp[i][0]=used[i+1],dp[i][1]=0;
            continue;
        }
        else{
            dp[i][0]=max({dp[i-1][0]-used[i],dp[i-2][1]+used[i-1],dp[i-2][0]})+used[i+1];
            dp[i][1]=max(dp[i-1][1],dp[i-1][0]);
        }
    }
    return max(dp[N][1],dp[N][0]);
  }
  else if(cntt==M){
    for(ll i=0; i<M; i++){
        if(X[i]==0) sum1+=W[i];
        else sum2+=W[i];
    }
    return max(sum1,sum2);
  }
}

컴파일 시 표준 에러 (stderr) 메시지

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

• Subtask 1: 3 / 3
• Subtask 2: 0 / 6
• Subtask 3: 9 / 9
• Subtask 4: 0 / 14
• Subtask 5: 0 / 21
• Subtask 6: 0 / 17
• Subtask 7: 0 / 14
• Subtask 8: 0 / 16