제출 #649180

#제출 시각아이디문제언어결과실행 시간메모리
649180teokakabadze메기 농장 (IOI22_fish)C++17
0 / 100
1076 ms116944 KiB
#include<bits/stdc++.h>
using namespace std;

#define f first
#define s second
#define pb push_back
typedef long long ll;

ll i, j, k, t, l, x, y, w, ans;
vector<ll> v[100005];
map<pair<ll, ll>, ll> m, p, dp;
set<ll> s;
set<ll> :: iterator it, it1;

ll max_weights(int N, int M, std::vector<int> X, std::vector<int> Y, std::vector<int> W) 
{
    for(i = 0; i < M; i++)
    {
        X[i]++; Y[i]++;
        v[X[i] + 1].pb(Y[i]);
        m[{X[i], Y[i]}] = W[i];
    }
    for(i = 1; i <= N + 1; i++)
    {
        v[i].pb(0), v[i].pb(N + 1);
        sort(v[i].begin(), v[i].end());
        /*for(auto j : v[i])
        cout << j << ' ';
        cout << endl; */
    } 
    for(i = 1; i <= N; i++)
    {
        for(j = 0; j < v[i - 1].size(); j++)
        if(v[i - 1][j]) s.insert(v[i - 1][j] - 1);
        for(j = 0; j < v[i].size() - 1; j++)
        s.insert(v[i][j]);
        for(j = 0; j < v[i + 1].size() - 1; j++)
        s.insert(v[i + 1][j]);
        s.insert(N);
        for(it = s.begin(); it != s.end(); it++)
        {
            p[{i, *it}] = 0;
            if(m[{i,*it}]) p[{i, *it}] = m[{i, *it}]; 
            if(it != s.begin())
            {
                it1 = it;
                it1--;
                p[{i, *it}] += p[{i, *it1}];
            }
           // cout << *it << ' ' << p[{i, *it}] << ' ';
        }
        //cout << endl;
        s.clear();
    } 
    for(i = 2; i <= N; i++)
    {
        k = 0; t = 0;
        for(j = 0; j < v[i].size() - 1; j++)
        {
            if(j && v[i][j] == v[i][j - 1]) continue;
            if(!j)
            {
                //cout << p[{i, v[i - 1][l + 1] - 1}] << endl;
                for(l = 0; l < v[i - 1].size() - 1; l++)
                dp[{i, j}] = max(dp[{i, j}], dp[{i - 1, l}] + p[{i, v[i - 1][l + 1] - 1}]);
            }
            else
            {
                while(k < v[i - 1].size() && v[i - 1][k] < v[i][j])
                {
                    if(!v[i - 1][k])
                    {
                        if(i == 2) { dp[{i, j}] = max(dp[{i, j}], p[{1, v[i][j]}]); k++; continue; }
                        while(t < v[i - 2].size() && v[i - 2][t] < v[i][j])
                        {
                            dp[{i, j}] = max(dp[{i, j}], dp[{i - 2, t}] + p[{i - 1, v[i][j]}]);
                            t++;
                        }
                    }
                    else
                    dp[{i, j}] = max(dp[{i, j}], dp[{i - 1, k}] + p[{i - 1, v[i][j]}] - p[{i - 1, v[i - 1][k]}]);
                    k++;
                }
                if(j - 1) dp[{i, j}] = max(dp[{i, j}], dp[{i, j - 1}] + p[{i - 1, v[i][j]}] - p[{i - 1, v[i][j - 1]}]);
                //cout << p[{i - 1, v[i][j]}] << endl;
            }
            //cout << i << ' ' << v[i][j] << ' ' << dp[{i, j}] << endl;
        }
    } 
    for(i = 0; i < v[N].size() - 1; i++)
    ans = max(ans, dp[{N, i}]); 
    return ans; 
}

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

fish.cpp: In function 'll max_weights(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
fish.cpp:33:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   33 |         for(j = 0; j < v[i - 1].size(); j++)
      |                    ~~^~~~~~~~~~~~~~~~~
fish.cpp:35:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   35 |         for(j = 0; j < v[i].size() - 1; j++)
      |                    ~~^~~~~~~~~~~~~~~~~
fish.cpp:37:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   37 |         for(j = 0; j < v[i + 1].size() - 1; j++)
      |                    ~~^~~~~~~~~~~~~~~~~~~~~
fish.cpp:58:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   58 |         for(j = 0; j < v[i].size() - 1; j++)
      |                    ~~^~~~~~~~~~~~~~~~~
fish.cpp:64:30: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   64 |                 for(l = 0; l < v[i - 1].size() - 1; l++)
      |                            ~~^~~~~~~~~~~~~~~~~~~~~
fish.cpp:69:25: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   69 |                 while(k < v[i - 1].size() && v[i - 1][k] < v[i][j])
      |                       ~~^~~~~~~~~~~~~~~~~
fish.cpp:74:33: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   74 |                         while(t < v[i - 2].size() && v[i - 2][t] < v[i][j])
      |                               ~~^~~~~~~~~~~~~~~~~
fish.cpp:90:18: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   90 |     for(i = 0; i < v[N].size() - 1; i++)
      |                ~~^~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...