Submission #116401

#	Time	Username	Problem	Language	Result	Execution time	Memory
116401		johutha	Aliens (IOI16_aliens)	C++14	0 / 100	2 ms	384 KiB

aliens

#include "aliens.h"
#include <vector>
#include <iostream>
#include <queue>

using namespace std;

long long take_photos(int n, int m, int k, vector<int> r, vector<int> c) {
    bool b = true;
    for (int i = 0; i < n; i++)
    {
        if (r[i] != c[i]) b = false;
    }
    if (!b)
    {
        int res = 0;
        for (int i = 0; i < n; i++)
        {
            if (r[i] < c[i]) swap(r[i], c[i]);
        }
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < m; j++)
            {
                bool on = false;
                for (int l = 0; l < n; l++)
                {
                    if (i <= r[l] && i >= c[l] && j <= r[l] && j >= c[l]) {on = true; break;}
                }
                res += on;
            }
        }
        return res;
    }
    else
    {
        priority_queue<int> pq;
        for (int i = 0; i < n; i++)
        {
            pq.push(-1);
        }
        for (int i = 0; i < n - k; i++)
        {
            int i1 = -pq.top();
            pq.pop();
            int i2 = -pq.top();
            pq.pop();
            pq.push(i1 + i2);
        }
        int res = 0;
        while (pq.size() > 0)
        {
            res += (pq.top() * pq.top());
            pq.pop();
        }
        return res;
    }
}

• Subtask 1: 0 / 4
• Subtask 2: 0 / 12
• Subtask 3: 0 / 9
• Subtask 4: 0 / 16
• Subtask 5: 0 / 19
• Subtask 6: 0 / 40