結果

問題 No.1536 仕切り直し
ユーザー inksamuraiinksamurai
提出日時 2021-06-06 19:37:28
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 45 ms / 2,000 ms
コード長 1,975 bytes
コンパイル時間 1,126 ms
コンパイル使用メモリ 108,888 KB
実行使用メモリ 56,192 KB
最終ジャッジ日時 2024-05-04 00:50:26
合計ジャッジ時間 1,853 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 4 ms
7,424 KB
testcase_04 AC 17 ms
21,376 KB
testcase_05 AC 32 ms
39,680 KB
testcase_06 AC 5 ms
8,704 KB
testcase_07 AC 5 ms
8,704 KB
testcase_08 AC 15 ms
21,120 KB
testcase_09 AC 19 ms
25,472 KB
testcase_10 AC 45 ms
56,192 KB
testcase_11 AC 11 ms
15,360 KB
testcase_12 AC 34 ms
41,856 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cmath>
#include <deque>
#include <algorithm>
#include <iterator>
#include <list>
#include <tuple>
#include <map>
#include <unordered_map>
#include <queue>
#include <set>
#include <unordered_set>
#include <stack>
#include <string>
#include <vector>
#include <fstream>
#include <iostream>
#include <functional>
#include <numeric>
#include <iomanip> 
#include <stdio.h>
//eolibraries
#define lnf 3999999999999999999
#define inf 3999999999999999999
#define PI 3.14159265359
#define endl "\n"
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define ll long long
#define all(c) (c).begin(),(c).end()
#define sz(c) (ll)(c).size()
#define mkp(a,b) make_pair(a,b)
#define make_unique(a) sort(all(a)),a.erase(unique(all(a)),a.end())
#define rsz(a,n) a.resize(n)
#define pii pair <ll,ll>
#define rep(i,n) for(ll i = 0 ; i < n ; i++) 
#define drep(i,n) for(ll i = n-1 ; i >= 0 ; i--)
#define crep(i,x,n) for(ll i = x ; i < n ; i++)
#define vi vector <ll> 
#define vec(s) vector<s>
#define rsz(a,n) a.resize(n)
#define rszv(a,n,v) a.resize(n,v)
#define fcin ios_base::sync_with_stdio(false),cin.tie(0),cout.tie(0);
//eodefine
 
const ll max_n = 2005;
 
using namespace std;
ll n,m;
vi a;
pii dp[max_n][max_n];

int main(){
fcin;
	cin>>n>>m;
	n++;
	a.resize(n,0);
	
	rep(i,n) {
		cin>>a[i];
	}
	
	a.pb(0);

	rep(i,n) {
		rep(j,m+2) dp[i][j]={-inf,-1};
	}
	
	dp[0][0]={0,0};

	rep(i,n) {
		rep(j,m+1) {
			dp[i+1][j].fi = dp[i][j].fi + (j%2!=0 ? -a[i] : a[i]);
			dp[i+1][j].se = 0;
		}
		rep(j,m+1) {
			if(dp[i+1][j+1].fi < dp[i][j].fi + ((j+1)%2!=0?-a[i]:a[i])) {
				dp[i+1][j+1].fi = dp[i][j].fi + ((j+1)%2!=0?-a[i]:a[i]);
				dp[i+1][j+1].se = 1;
			}
		}
	} 
	
	ll mx=0,nj=-1;

	rep(j,m+1) {
		if(dp[n][j].fi>mx) {
			mx=dp[n][j].fi;
			nj=j;
		}
	}

	ll i=n,j=nj;
	vi ans;

	while(i>0) {
		if(dp[i][j].se == 1) {ans.pb(i-1); j--;}
		i--;
	}


	while(sz(ans)<m) ans.pb(n-1);

	for(auto x : ans) cout << x << "\n";

/*
*/
 
    return 0;   
}
0