이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <iostream>
#include <vector>
#include <set>
#include <cmath>
#include <algorithm>
#include <map>
#include <queue>
#include <deque>
#include <string>
#include <unordered_set>
#define Bekabot ios_base::sync_with_stdio(NULL);cin.tie(0);cout.tie(0);
#define fopen(s) freopen(s".in","r",stdin);freopen(s".out","w",stdout)
#define all(x) (x).begin(), (x).end()
#define len(x) (int)x.size()
#define pb push_back
#define pf push_front
#define popb pop_back
#define popf pop_front
#define elif else if
#define F first
#define S second
#define int long long
#define pii pair<int,int>
#define lb(v,x) (int)(lower_bound(all(v),x)-v.begin())
#define ub(v,x) (int)(upper_bound(all(v),x)-v.begin())
const int N = 1e5 + 78 , inf = 1e18 , M = 1e9 + 7;
using namespace std;
string trans(int n , int k){
string s = "";
while(n != 0){
int d = n % k;
if(d < 10){
s = to_string(d) + s;
}
else{
s = char(d + 'A' - 10) + s;
}
n /= k;
}
return s;
}
int pref[N] , suff[N];
void solve(){
int n , k;
cin >> n >> k;
int a[n + 1];
int x = 0;
vector<int> v1;
for(int i = 1 ; i <= n ; i++)cin >> a[i];
for(int i = 1 ; i <= n ; i++)pref[i] = pref[i - 1] + a[i];
for(int i = n ; i >= 1 ; i--)suff[i] = suff[i + 1] + a[i];
for(int i = 10000 ; i >= 0 ; i--){
for(int l = 1 ; l < n ; l++){
int s = 0 , c = 0;
vector<int> v;
c += pref[l] * suff[l + 1];
v.pb(l);
for(int j = l + 1 ; j < n ; j++){
s += a[j];
if(s >= i and len(v) < k){
c += s * suff[j + 1];
s = 0;
v.pb(j);
}
}
//if(i == 4)cout << s << ' ';
if(len(v) == k and c > x){
x = c;
v1.clear();
for(auto it : v)v1.pb(it);
}
}
}
cout << x << '\n';
for(auto it : v1)cout << it << ' ';
}
signed main(){
Bekabot
int t = 1;
//cin >> t;
while(t--){
solve();
}
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |