Submission #499631

#TimeUsernameProblemLanguageResultExecution timeMemory
499631pragmatistSplit the sequence (APIO14_sequence)C++17
11 / 100
9 ms1868 KiB
#include <bits/stdc++.h>                            
 
#define pb push_back
#define sz(v) (int)v.size()
#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
#define x first
#define y second
#define int long long
#define nl "\n"
 
using namespace std;

typedef long long ll;
typedef pair<long long, long long> pll;
typedef pair <ll, ll> pii;

const int N = (int)1e5 + 7;
const int M = (int)7e6 + 7;
const ll MOD = (ll)1e9 + 7;                    
const int inf = (int)1e9 + 7;
const ll INF = (ll)3e18 + 7;

pii dir[] = {{-1, -1}, {1, 1}, {-1, 1}, {1, -1}};

int n, k, a[N], p[N];

void solve() {        
	cin >> n >> k;
	for(int i = 0; i < n; ++i) {
		cin >> a[i];
		if(i) p[i] = p[i - 1];
		p[i] += a[i]; 
	}
	int ans = -INF, id = 0;
	for(int mask = 0; mask < (1 << n); ++mask) {
		int x = __builtin_popcount(mask); 
		if(x != k || (mask & (1 << (n - 1)))) continue;
		int l = 0, cur = 0;
		
		for(int i = 0; i < n; ++i) {
			if(mask >> i & 1) {
				cur += 1ll * (p[i] - (l == 0 ? 0 : p[l - 1])) * (p[n-1] - p[i]);
				l = i + 1;					
			}
		}                    
		if(cur > ans) {
			ans = cur;
			id = mask;
		}	
	}                                                           
	cout << ans << nl;			
	for(int i = 0; i < n; ++i) if(id >> i & 1) cout << i + 1 << ' ';
}

signed main() {                   
	ios_base::sync_with_stdio(NULL);
    cin.tie(0);
    cout.tie(0);      
   	int test = 1;
	//cin >> test;
	for(int i = 1; i <= test; ++i) {
        //cout << "Case " << i << ": ";
        solve();
    }
    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...