Submission #483158

#	Time	Username	Problem	Language	Result	Execution time	Memory
483158		kawaii	Cake 3 (JOI19_cake3)	C++14	0 / 100	0 ms	332 KiB

cake3

#include <bits/stdc++.h>
using namespace std;
#define int long long

int n, m, v[1000005], c[1000005], ans = 0;
pair<int, int> dp[1000005];

struct num{
    int V, C;
    num();
    num(int A, int B){
        V = A, C = B;
    }
};

bool operator < (num a, num b){
    return a.C < b.C;
}

vector<num> vec;
deque<num> dq;

void solve(){  
    sort(vec.begin(), vec.end()); 
    for(auto i: vec){
        ans += i.V;
        dq.push_back(i);
    }
    ans -= 2 * (vec.back().C - vec[0].C); 
    while(dq.size() > m){
        num fr = dq.front(), ba = dq.back();
        dq.pop_front();
        int if1 = -fr.V + abs(fr.C - dq.front().C) + abs(fr.C - dq.back().C) - abs(dq.back().C - dq.front().C);
        dq.push_front(fr);
        dq.pop_back();
        int if2 = -ba.V + abs(ba.C - dq.back().C) + abs(ba.C - dq.front().C) - abs(dq.back().C - dq.front().C);;
        dq.push_back(ba);
        if(if1 > if2) dq.pop_front();
        else dq.pop_back();
        ans += max(if1, if2);  
    }
    cout << ans;
}

signed main(){
    if (ifstream("text.txt"))
    {
        freopen("text.txt", "r", stdin);
        #ifdef CHECKING
            freopen(taskname".out", "w", stdout);
        #else
            //\freopen("text.out", "w", stdout);
        #endif //CHECKING
    }
    //freopen("a.inp", "r", stdin);
    //freopen("a.out", "w", stdout);
    ios_base::sync_with_stdio(NULL); cin.tie(nullptr); cout.tie(nullptr);
    cin >> n >> m;
    for(int i = 1; i <= n; i++){
        cin >> v[i] >> c[i];  
        vec.push_back(num(v[i], c[i]));
    }
    solve();
}

Compilation message (stderr)

cake3.cpp: In function 'void solve()':
cake3.cpp:30:21: warning: comparison of integer expressions of different signedness: 'std::deque<num>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   30 |     while(dq.size() > m){
      |           ~~~~~~~~~~^~~
cake3.cpp: In function 'int main()':
cake3.cpp:48:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   48 |         freopen("text.txt", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 19
• Subtask 3: 0 / 76