/*#pragma GCC optimize("O3")
#pragma GCC target ("avx2")
#pragma GCC optimize("Ofast")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#pragma GCC optimize("unroll-loops")*/
#include <bits/stdc++.h>
#pragma GCC optimize ("unroll-loops,Ofast,O3")
#pragma GCC target("avx,avx2,fma")
//#define file(data) freopen(data".in", "r", stdin); freopen(data".out", "w", stdout);
#define pb push_back
//#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define all(data) data.begin() , data.end()
#define endl '\n'
//freopen("nenokku_easy.in", "r", stdin);
//freopen("nenokku_easy.out", "w", stdout);
#define int long long
#define pii pair < int, int >
#define pll pair < long long, long long >
using namespace std;
typedef long long ll;
const int N = 2e5 + 5;
const int M = 305;
const int mod = 1e9 + 7;
int n, k, a[N], res, use;
vector < int > dv;
void rec(int l, int r, int last) {
if(l >= r) return;
last++;
while(last < dv.size() && dv[last] < l) last++;
if(last == dv.size() || r <= dv[last]) return;
int lef = 0, righ = 0;
for(int i = l; i <= dv[last]; i++) {
lef += a[i];
}
for(int i = dv[last] + 1; i <= r; i++) {
righ += a[i];
}
res += lef * righ;
++use;
rec(l, dv[last], last);
rec(dv[last] + 1, r, last);
}
main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
//file("pieaters");
cin >> n >> k;
vector < int > br;
for(int i = 1; i <= n; i++) {
br.pb(i);
cin >> a[i];
}
int ans = 0;
vector < int > path;
do {
for(int i = 0; i < k; i++) dv.pb(br[i]);
res = 0;
rec(1, n, -1);
if(res >= ans && use == k) {
path = dv;
ans = res;
}
use = 0;
dv.clear();
} while(next_permutation(all(br)));
reverse(all(path));
cout << ans << endl;
for(int i = 0; i < path.size(); i++) {
cout << path[i] << " ";
}
}
Compilation message
sequence.cpp: In function 'void rec(long long int, long long int, long long int)':
sequence.cpp:33:16: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
33 | while(last < dv.size() && dv[last] < l) last++;
| ~~~~~^~~~~~~~~~~
sequence.cpp:34:13: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
34 | if(last == dv.size() || r <= dv[last]) return;
| ~~~~~^~~~~~~~~~~~
sequence.cpp: At global scope:
sequence.cpp:48:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
48 | main() {
| ^~~~
sequence.cpp: In function 'int main()':
sequence.cpp:74:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
74 | for(int i = 0; i < path.size(); i++) {
| ~~^~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
312 KB |
contestant found the optimal answer: 108 == 108 |
2 |
Correct |
138 ms |
204 KB |
contestant found the optimal answer: 999 == 999 |
3 |
Correct |
0 ms |
204 KB |
contestant found the optimal answer: 0 == 0 |
4 |
Correct |
274 ms |
296 KB |
contestant found the optimal answer: 1542524 == 1542524 |
5 |
Correct |
279 ms |
312 KB |
contestant found the optimal answer: 4500000000 == 4500000000 |
6 |
Correct |
105 ms |
204 KB |
contestant found the optimal answer: 1 == 1 |
7 |
Correct |
106 ms |
300 KB |
contestant found the optimal answer: 1 == 1 |
8 |
Correct |
118 ms |
300 KB |
contestant found the optimal answer: 1 == 1 |
9 |
Correct |
173 ms |
204 KB |
contestant found the optimal answer: 100400096 == 100400096 |
10 |
Correct |
185 ms |
300 KB |
contestant found the optimal answer: 900320000 == 900320000 |
11 |
Correct |
196 ms |
296 KB |
contestant found the optimal answer: 3698080248 == 3698080248 |
12 |
Correct |
196 ms |
300 KB |
contestant found the optimal answer: 3200320000 == 3200320000 |
13 |
Correct |
23 ms |
316 KB |
contestant found the optimal answer: 140072 == 140072 |
14 |
Correct |
1 ms |
312 KB |
contestant found the optimal answer: 376041456 == 376041456 |
15 |
Correct |
210 ms |
300 KB |
contestant found the optimal answer: 805 == 805 |
16 |
Correct |
175 ms |
316 KB |
contestant found the optimal answer: 900189994 == 900189994 |
17 |
Correct |
199 ms |
300 KB |
contestant found the optimal answer: 999919994 == 999919994 |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
2083 ms |
204 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
2045 ms |
204 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
2043 ms |
332 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
2059 ms |
588 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
2078 ms |
1996 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |