fork download

copy

#include <bits/stdc++.h>
using namespace std;
signed main()
{
    int t;
    cin >> t;
    while (t--)
    {
        int n, k;
        cin >> n >> k;
        vector<int> arr(n);
        for (int i = 0; i < n; ++i)
        {
            cin >> arr[i];
        }
        vector<int> diffs(n - 1);
        for (int i = 1; i < n; ++i)
        {
            diffs[i - 1] = arr[i] - arr[i - 1];
        }
        vector<int> next(n - 1), prev(n - 1);
        for (int i = 0; i < n - 1; ++i)
        {
            next[i] = i + 1;
            prev[i] = i - 1;
        }
        next[n - 2] = -1;
        prev[0] = -1;
        set<pair<int, int>> s;
        for (int i = 0; i < n - 1; ++i)
        {
            s.insert({diffs[i], i});
        }
        int sol = 0;
        for (int i = 0; i < k; ++i)
        {
            auto it = s.begin();
            int pos = it->second;
            s.erase(it);
            int l = prev[pos];
            int r = next[pos];
            if (l != -1)
            {
                s.erase({diffs[l], l});
            }
            if (r != -1)
            {
                s.erase({diffs[r], r});
            }
            sol += diffs[pos];
            diffs[pos] = (l != -1 ? diffs[l] : 0) + (r != -1 ? diffs[r] : 0) - diffs[pos];
            if (l != -1 && r != -1)
            {
                s.insert({diffs[pos], pos});
                prev[pos] = prev[l];
                next[pos] = next[r];
            }
            else
            {
                pos = -1;
            }
            if (l != -1 && prev[l] != -1)
            {
                next[prev[l]] = pos;
            }
            if (r != -1 && next[r] != -1)
            {
                prev[next[r]] = pos;
            }
        }
        cout << sol << "\n";
    }
 
}

Main.java:1: error: illegal character: '#'
#include <bits/stdc++.h>
^
Main.java:1: error: class, interface, or enum expected
#include <bits/stdc++.h>
         ^
Main.java:3: error: class, interface, or enum expected
signed main()
^
Main.java:6: error: class, interface, or enum expected
    cin >> t;
    ^
Main.java:7: error: class, interface, or enum expected
    while (t--)
    ^
Main.java:10: error: class, interface, or enum expected
        cin >> n >> k;
        ^
Main.java:11: error: class, interface, or enum expected
        vector<int> arr(n);
        ^
Main.java:12: error: class, interface, or enum expected
        for (int i = 0; i < n; ++i)
        ^
Main.java:12: error: class, interface, or enum expected
        for (int i = 0; i < n; ++i)
                        ^
Main.java:12: error: class, interface, or enum expected
        for (int i = 0; i < n; ++i)
                               ^
Main.java:15: error: class, interface, or enum expected
        }
        ^
Main.java:17: error: class, interface, or enum expected
        for (int i = 1; i < n; ++i)
        ^
Main.java:17: error: class, interface, or enum expected
        for (int i = 1; i < n; ++i)
                        ^
Main.java:17: error: class, interface, or enum expected
        for (int i = 1; i < n; ++i)
                               ^
Main.java:20: error: class, interface, or enum expected
        }
        ^
Main.java:22: error: class, interface, or enum expected
        for (int i = 0; i < n - 1; ++i)
        ^
Main.java:22: error: class, interface, or enum expected
        for (int i = 0; i < n - 1; ++i)
                        ^
Main.java:22: error: class, interface, or enum expected
        for (int i = 0; i < n - 1; ++i)
                                   ^
Main.java:25: error: class, interface, or enum expected
            prev[i] = i - 1;
            ^
Main.java:26: error: class, interface, or enum expected
        }
        ^
Main.java:28: error: class, interface, or enum expected
        prev[0] = -1;
        ^
Main.java:29: error: class, interface, or enum expected
        set<pair<int, int>> s;
        ^
Main.java:30: error: class, interface, or enum expected
        for (int i = 0; i < n - 1; ++i)
        ^
Main.java:30: error: class, interface, or enum expected
        for (int i = 0; i < n - 1; ++i)
                        ^
Main.java:30: error: class, interface, or enum expected
        for (int i = 0; i < n - 1; ++i)
                                   ^
Main.java:33: error: class, interface, or enum expected
        }
        ^
Main.java:35: error: class, interface, or enum expected
        for (int i = 0; i < k; ++i)
        ^
Main.java:35: error: class, interface, or enum expected
        for (int i = 0; i < k; ++i)
                        ^
Main.java:35: error: class, interface, or enum expected
        for (int i = 0; i < k; ++i)
                               ^
Main.java:38: error: class, interface, or enum expected
            int pos = it->second;
            ^
Main.java:39: error: class, interface, or enum expected
            s.erase(it);
            ^
Main.java:40: error: class, interface, or enum expected
            int l = prev[pos];
            ^
Main.java:41: error: class, interface, or enum expected
            int r = next[pos];
            ^
Main.java:42: error: class, interface, or enum expected
            if (l != -1)
            ^
Main.java:45: error: class, interface, or enum expected
            }
            ^
Main.java:49: error: class, interface, or enum expected
            }
            ^
Main.java:51: error: class, interface, or enum expected
            diffs[pos] = (l != -1 ? diffs[l] : 0) + (r != -1 ? diffs[r] : 0) - diffs[pos];
            ^
Main.java:52: error: class, interface, or enum expected
            if (l != -1 && r != -1)
            ^
Main.java:55: error: class, interface, or enum expected
                prev[pos] = prev[l];
                ^
Main.java:56: error: class, interface, or enum expected
                next[pos] = next[r];
                ^
Main.java:57: error: class, interface, or enum expected
            }
            ^
Main.java:61: error: class, interface, or enum expected
            }
            ^
Main.java:65: error: class, interface, or enum expected
            }
            ^
Main.java:69: error: class, interface, or enum expected
            }
            ^
Main.java:72: error: class, interface, or enum expected
    }
    ^
45 errors

Standard output is empty