제출 #239758

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
239758		Vimmer	Go (COCI18_go)	C++14	100 / 100	191 ms	504 KiB

go

#include <bits/stdc++.h>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>

//#pragma GCC optimize("unroll-loops")
//#pragma GCC optimize("-O3")
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("fast-math")
//#pragma GCC optimize("no-stack-protector")

#define F first
#define S second
#define sz(x) int(x.size())
#define pb push_back
#define N 101001
#define M ll(1e9 + 7)
#define inf 1e9 + 1e9


using namespace std;
//using namespace __gnu_pbds;

typedef long double ld;
typedef long long ll;
typedef short int si;
typedef array <int, 3> a3;

//typedef tree <int, null_type, less_equal<int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;


vector <pair <int, int> > g[1005];

int main()
{
    //freopen("input.txt", "r", stdin); //freopen("output4.txt", "w", stdout);

    ios_base::sync_with_stdio(0); istream::sync_with_stdio(0); cin.tie(0); cout.tie(0);

    int n, k, m;

    cin >> n >> k >> m;

    for (int i = 0; i < m; i++) {int x, y, z; cin >> x >> y >> z; g[x].pb({z, y});}

    for (int i = 1; i <= n; i++) sort(g[i].begin(), g[i].end());

    int ans = 0;

    for (int i = 1; i <= k; i++)
    {
        int sum = 0, tmr = k - i + 1;

        for (int j = i; j <= k; j++, tmr--)
            for (auto it : g[j])
              if (tmr <= it.F) sum += it.S;

        tmr = (k - i + 1) * 2;

        for (int j = k + 1; j <= n; j++, tmr++)
            for (auto it : g[j])
              if (tmr <= it.F) sum += it.S;

        ans = max(ans, sum);
    }

    for (int i = k; i <= n; i++)
    {
        int sum = 0;

        int tmr = i - k + 1;

        for (int j = i; j >= k; j--, tmr--)
            for (auto it : g[j])
              if (tmr <= it.F) sum += it.S;

        tmr = (i - k + 1) * 2;

        for (int j = k - 1; j >= 0; j--, tmr++)
            for (auto it : g[j])
              if (tmr <= it.F) sum += it.S;

        ans = max(ans, sum);
    }

    for (int i = 1; i <= k; i++)
        for (int j = k + 1; j <= n; j++)
        {
            int tmr = 1, sum = 0;

            for (int u = k; u <= j; u++, tmr++)
                for (auto it : g[u])
                  if (tmr <= it.F) sum += it.S;

            tmr += j - k;

            for (int u = k - 1; u >= i; u--, tmr++)
                for (auto it : g[u])
                  if (tmr <= it.F) sum += it.S;

            tmr += j - i;

            for (int u = j + 1; u <= n; u++, tmr++)
                for (auto it : g[u])
                  if (tmr <= it.F) sum += it.S;

            ans = max(ans, sum);
        }

    cout << ans << endl;
}

• Subtask 1: 100 / 100