結果

問題 No.324 落ちてた閉路グラフ
ユーザー srup٩(๑`н´๑)۶
提出日時 2016-08-31 21:11:39
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 2,840 bytes
コンパイル時間 512 ms
コンパイル使用メモリ 64,424 KB
実行使用メモリ 145,312 KB
最終ジャッジ日時 2024-11-14 13:37:01
合計ジャッジ時間 4,926 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 31 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstdio>
typedef long long ll;
using namespace std;
#define rep(i,n) for(int i=0;i<(n);i++)

int n, m;
//dp1[i][j][k] := (0番目の頂点を選ばず、)i番目までを考えて、
//選んだ頂点がj個でi番目の頂点をk=1の時選び、k=0の時選んでいない
int dp1[3010][3010][2];
//dp1[i][j][k] := (0番目の頂点を選び、)i番目までを考えて、
//選んだ頂点がj個でi番目の頂点をk=1の時選び、k=0の時選んでいない
int dp2[3010][3010][2];
const int INF = 1e9;
int main(void){
	cin >> n >> m;
	vector<int> w(n);
	rep(i, n) cin >> w[i];

	rep(i, 3010)rep(j, 3010)rep(k, 2){
		dp1[i][j][k] = dp2[i][j][k] = -INF;
	}
	dp1[0][0][0] = 0;
	for (int i = 0; i < n; ++i){
		for (int j = 0; j <= i + 1; ++j){
			for (int k = 0; k < 2; ++k){
				if(dp1[i][j][k] == -INF)continue;
				if(k == 0){//i番目の頂点を選んでいない
					dp1[i + 1][j + 1][1] = max(dp1[i][j][k], dp1[i + 1][j + 1][1]);//i+1番目を選ぶ
					dp1[i + 1][j][0] = max(dp1[i][j][k], dp1[i + 1][j][0]);//i+1番目を選ばない
				}else{//i番目の頂点を選んでいる
					dp1[i + 1][j + 1][1] = max(dp1[i][j][k] + w[i], dp1[i + 1][j + 1][1]);//i+1番目を選ぶ
					dp1[i + 1][j][0] = max(dp1[i][j][k], dp1[i + 1][j][0]);//i+1番目を選ばない
				}
			}
		}
	}

	dp2[0][1][1] = 0;
	for (int i = 0; i < n - 1; ++i){
		for (int j = 0; j <= i + 1; ++j){
			for (int k = 0; k < 2; ++k){
				if(dp2[i][j][k] == -INF)continue;
				if(k == 0){//i番目の頂点を選んでいない
					dp2[i + 1][j][0] = max(dp2[i][j][k], dp2[i + 1][j][0]);//i+1番目を選ばない
					// printf("2 dp2[%d][%d][0] = %d\n", i + 1, j, dp2[i + 1][j][0]);
					if(i == n - 2){
						dp2[i + 1][j + 1][1] = max(dp2[i][j][k] + w[i + 1], dp2[i + 1][j + 1][1]);
					}else{
						dp2[i + 1][j + 1][1] = max(dp2[i][j][k], dp2[i + 1][j + 1][1]);//i+1番目を選ぶ
						// printf("1 dp2[%d][%d][1] = %d\n", i + 1, j + 1, dp2[i + 1][j + 1][1]);
					}
				}else{//i番目の頂点を選んでいる
					dp2[i + 1][j][0] = max(dp2[i][j][k], dp2[i + 1][j][0]);//i+1番目を選ばない
					// printf("3 dp2[%d][%d][0] = %d\n", i + 1, j, dp2[i + 1][j][0]);

					if(i == n - 2){//0番目の頂点を選んでいるので
						dp2[i + 1][j + 1][1] = max(dp2[i][j][k] + w[i] + w[i + 1], dp2[i + 1][j + 1][1]);
						// printf("4 dp2[%d][%d][1] = %d\n", i + 1, j + 1, dp2[i + 1][j + 1][1]);

					}else{
						dp2[i + 1][j + 1][1] = max(dp2[i][j][k] + w[i], dp2[i + 1][j + 1][1]);//i+1番目を選ぶ
						// printf("5 dp2[%d][%d][1] = %d\n", i + 1, j + 1, dp2[i + 1][j + 1][1]);

					}
				}
			}
		}
	}

	int ans = 0;
	rep(i, n){
		rep(j, 2){
			ans = max(ans, dp1[n - 1][m][j]);
			ans = max(ans, dp2[n - 1][m][j]);
		}
	}
	printf("%d\n", ans);
	return 0;
}
0