結果

問題 No.324 落ちてた閉路グラフ
ユーザー furafura
提出日時 2020-07-12 03:02:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 102 ms / 5,000 ms
コード長 1,042 bytes
コンパイル時間 1,951 ms
コンパイル使用メモリ 196,436 KB
最終ジャッジ日時 2025-01-11 19:52:41
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 34
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:10:23: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   10 |         int n,m; scanf("%d%d",&n,&m);
      |                  ~~~~~^~~~~~~~~~~~~~
main.cpp:12:23: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   12 |         rep(i,n) scanf("%d",&a[i]);
      |                  ~~~~~^~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

#define rep(i,n) for(int i=0;i<(n);i++)

using namespace std;

const int INF=1<<29;

int main(){
	int n,m; scanf("%d%d",&n,&m);
	vector<int> a(n);
	rep(i,n) scanf("%d",&a[i]);

	if(m==0) return puts("0"),0;

	// 頂点 0 を選ばない
	static int dp1[3000][3001][2];
	rep(i,n) rep(j,m+1) rep(k,2) dp1[i][j][k]=-INF;
	dp1[0][0][0]=0;
	for(int i=1;i<n;i++){
		rep(j,m+1){
			// 頂点 i を選ばない
			dp1[i][j][0]=max(dp1[i-1][j][0],dp1[i-1][j][1]);
			// 頂点 i を選ぶ
			if(j<m) dp1[i][j+1][1]=max(dp1[i-1][j][0],dp1[i-1][j][1]+a[i-1]);
		}
	}

	// 頂点 0 を選ぶ
	static int dp2[3000][3001][2];
	rep(i,n) rep(j,m+1) rep(k,2) dp2[i][j][k]=-INF;
	dp2[0][1][1]=0;
	for(int i=1;i<n;i++){
		rep(j,m+1){
			// 頂点 i を選ばない
			dp2[i][j][0]=max(dp2[i-1][j][0],dp2[i-1][j][1]);
			// 頂点 i を選ぶ
			if(j<m) dp2[i][j+1][1]=max(dp2[i-1][j][0],dp2[i-1][j][1]+a[i-1])+(i==n-1?a[n-1]:0);
		}
	}

	printf("%d\n",max({dp1[n-1][m][0],dp1[n-1][m][1],dp2[n-1][m][0],dp2[n-1][m][1]}));

	return 0;
}
0