#include #include #include #include #include #include #include // 内部定数 #define D_VTX_MAX 3005 // 最大頂点数 #define D_VAL_INI (-300 * D_VTX_MAX) // 初期値 // 内部変数 static FILE *szpFpI; // 入力 static int si1Val[D_VTX_MAX]; // 価値[頂点] static int siVCnt; // 頂点数 static int siSCnt; // 選択数 static int si2SumE[D_VTX_MAX][D_VTX_MAX]; // 価値合計[頂点][頂点数] - 1つ前の選択あり static int si2SumN[D_VTX_MAX][D_VTX_MAX]; // 価値合計[頂点][頂点数] - 1つ前の選択なし static int siMax; // 最大値 // 内部変数 - テスト用 #ifdef D_TEST static int siRes; static FILE *szpFpA; #endif // 大きい値を取得 int fGetLarge( int piVal1 // 値1 , int piVal2 // 値2 ) { if (piVal1 > piVal2) { return piVal1; } else { return piVal2; } } // 最大値 - セット int fSetMax( int piSel // 1つ目の選択 [0]:なし [1]:あり ) { int i, j; // 価値合計 - 初期化 for (i = 0; i < siVCnt; i++) { for (j = 0; j <= i + 1; j++) { si2SumE[i][j] = D_VAL_INI; si2SumN[i][j] = D_VAL_INI; } } // 価値合計 - 1つ目・2つ目 if (piSel == 0) { si2SumN[0][0] = 0; si2SumN[1][0] = 0; si2SumE[1][1] = 0; } else { si2SumE[0][1] = 0; si2SumN[1][1] = 0; si2SumE[1][2] = si1Val[1]; } // 価値合計 - 3つ目以降 for (i = 2; i < siVCnt; i++) { // 今回選択なし・前回選択なし for (j = 0; j < i; j++) { si2SumN[i][j] = fGetLarge(si2SumN[i][j], si2SumN[i - 1][j]); } // 今回選択なし・前回選択あり for (j = 1; j <= i; j++) { si2SumN[i][j] = fGetLarge(si2SumN[i][j], si2SumE[i - 1][j]); } // 今回選択あり・前回選択なし for (j = 1; j <= i; j++) { if (piSel == 0 || i < siVCnt - 1) { si2SumE[i][j] = fGetLarge(si2SumE[i][j], si2SumN[i - 1][j - 1]); } else { si2SumE[i][j] = fGetLarge(si2SumE[i][j], si2SumN[i - 1][j - 1] + si1Val[0]); } } // 今回選択あり・前回選択あり for (j = 2; j <= i + 1; j++) { if (piSel == 0 || i < siVCnt - 1) { si2SumE[i][j] = fGetLarge(si2SumE[i][j], si2SumE[i - 1][j - 1] + si1Val[i]); } else { si2SumE[i][j] = fGetLarge(si2SumE[i][j], si2SumE[i - 1][j - 1] + si1Val[i] + si1Val[0]); } } } // 最大値 - 更新 siMax = fGetLarge(siMax, si2SumN[siVCnt - 1][siSCnt]); siMax = fGetLarge(siMax, si2SumE[siVCnt - 1][siSCnt]); return 0; } // 実行メイン int fMain( int piTNo // テスト番号 1~ ) { int i; char lc1Buf[1024], lc1Out[1024]; // データ - 初期化 siMax = INT_MIN; // 最大値 // 入力 - セット #ifdef D_TEST sprintf(lc1Buf, ".\\Test\\T%d.txt", piTNo); szpFpI = fopen(lc1Buf, "r"); sprintf(lc1Buf, ".\\Test\\A%d.txt", piTNo); szpFpA = fopen(lc1Buf, "r"); siRes = 0; #else szpFpI = stdin; #endif // 頂点数・選択数 - 取得 fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d", &siVCnt, &siSCnt); // 価値[頂点] - 取得 for (i = 0; i < siVCnt; i++) { fscanf(szpFpI, "%d", &si1Val[i]); } fgets(lc1Buf, sizeof(lc1Buf), szpFpI); // 最大値 - セット fSetMax(0); fSetMax(1); // 結果 - セット sprintf(lc1Out, "%d\n", siMax); // 結果 - 表示 #ifdef D_TEST fgets(lc1Buf, sizeof(lc1Buf), szpFpA); if (strcmp(lc1Buf, lc1Out)) { siRes = -1; } #else printf("%s", lc1Out); #endif // 残データ有無 #ifdef D_TEST lc1Buf[0] = '\0'; fgets(lc1Buf, sizeof(lc1Buf), szpFpA); if (strcmp(lc1Buf, "")) { siRes = -1; } #endif // テストファイルクローズ #ifdef D_TEST fclose(szpFpI); fclose(szpFpA); #endif // テスト結果 #ifdef D_TEST if (siRes == 0) { printf("OK %d\n", piTNo); } else { printf("NG %d\n", piTNo); } #endif return 0; } int main() { #ifdef D_TEST int i; for (i = D_TEST_SNO; i <= D_TEST_ENO; i++) { fMain(i); } #else fMain(0); #endif return 0; }