#include #include #include #include #include #include #include // 内部定数 #define D_SEGT_CNT 131072 // セグメントツリーデータ数(2の累乗) #define D_SEGT_KIND 3 // セグメントツリーデータ種類数 #define D_SEGT_SUM 0 // セグメントツリーデータ種類 - 範囲取得 - かっこよさ合計 #define D_SEGT_MIN 1 // セグメントツリーデータ種類 - 範囲更新 - 最小車両 #define D_SEGT_MAX 2 // セグメントツリーデータ種類 - 範囲更新 - 最大車両 // 内部構造体 - セグメントツリー情報 typedef struct SegT { long long ml1Val[D_SEGT_KIND]; // 値 } SegT; // 内部変数 static FILE *szpFpI; // 入力 static SegT sz1SegT[D_SEGT_CNT * 2]; // セグメントツリー static int siSCNo; // セグメントツリー - 子の開始位置 // 内部変数 - テスト用 #ifdef D_TEST static int siRes; static FILE *szpFpA; static int siTNo; #endif // 出力 int fOut( char *pcpLine // 1行 ) { char lc1Buf[1024]; #ifdef D_TEST fgets(lc1Buf, sizeof(lc1Buf), szpFpA); if (strcmp(lc1Buf, pcpLine)) { siRes = -1; } #else printf("%s", pcpLine); #endif return 0; } // セグメントツリー - 子の開始位置 - セット int fSegTSetCSNo( int piCnt // 子データ数 ) { siSCNo = 1; while (siSCNo < piCnt) { siSCNo *= 2; } return 0; } // セグメントツリー - 親データセット int fSegTSetPData( ) { int i; // 最初の親番号範囲 int liPNo1 = siSCNo / 2; int liPNo2 = siSCNo - 1; // 作成 while (liPNo1 > 0) { for (i = liPNo1; i <= liPNo2; i++) { // 対象値を取得 long long llVal = sz1SegT[i * 2].ml1Val[D_SEGT_SUM] + sz1SegT[i * 2 + 1].ml1Val[D_SEGT_SUM]; // セット sz1SegT[i].ml1Val[D_SEGT_SUM] = llVal; } // 次の親番号範囲 liPNo2 = liPNo1 - 1; liPNo1 /= 2; } return 0; } // セグメントツリー - 範囲取得 long long fSegTGetRngMain( int piGetS // 取得範囲 - 開始 0~ , int piGetE // 取得範囲 - 終了 0~ , int piNNo // 現在位置 1~ , int piNowS // 現在範囲 - 開始 0~ , int piNowE // 現在範囲 - 終了 0~ ) { // 内包チェック if (piGetS <= piNowS && piNowE <= piGetE) { return sz1SegT[piNNo].ml1Val[D_SEGT_SUM]; } // 中間位置 int liCenter = (piNowS + piNowE) / 2; // 初期値 long long llVal = 0; // 左側 if (piGetS <= liCenter) { llVal = fSegTGetRngMain(piGetS, piGetE, piNNo * 2, piNowS, liCenter); } // 右側 if (piGetE >= liCenter + 1) { llVal += fSegTGetRngMain(piGetS, piGetE, piNNo * 2 + 1, liCenter + 1, piNowE); } return llVal; } long long fSegTGetRng( int piGetS // 取得範囲 - 開始 0~ , int piGetE // 取得範囲 - 終了 0~ ) { return fSegTGetRngMain(piGetS, piGetE, 1, 0, siSCNo - 1); } // セグメントツリー - 1データ加算 int fSegTUpOne( int piUpNo // 更新位置 0~ ) { // 子番号 int liCNo1 = siSCNo + piUpNo; int liCNo2; if (liCNo1 % 2 == 0) { liCNo2 = liCNo1 + 1; } else { liCNo2 = liCNo1 - 1; } // 加算 sz1SegT[liCNo1].ml1Val[D_SEGT_SUM]++; // 親を更新 while (1) { // 親番号 int liPNo = liCNo1 / 2; if (liPNo < 1) { break; } // 対象値を取得 long long llVal = sz1SegT[liCNo1].ml1Val[D_SEGT_SUM] + sz1SegT[liCNo2].ml1Val[D_SEGT_SUM]; // 更新 sz1SegT[liPNo].ml1Val[D_SEGT_SUM] = llVal; // 次の子番号 liCNo1 = liPNo; if (liCNo1 % 2 == 0) { liCNo2 = liCNo1 + 1; } else { liCNo2 = liCNo1 - 1; } } return 0; } // セグメントツリー - 範囲更新 int fSegTUpRngMain( int piKind // 種類 , int piUpS // 更新範囲 - 開始 0~ , int piUpE // 更新範囲 - 終了 0~ , int piUpVal // 更新値 , int piNNo // 現在位置 1~ , int piNowS // 現在範囲 - 開始 0~ , int piNowE // 現在範囲 - 終了 0~ ) { // 範囲外チェック if (piUpE < piNowS || piNowE < piUpS) { return 0; } // 範囲内なら更新 if (piUpS <= piNowS && piNowE <= piUpE) { sz1SegT[piNNo].ml1Val[piKind] = piUpVal; return 0; } // 更新済みの場合、子に展開しておく if (sz1SegT[piNNo].ml1Val[piKind] != 0) { sz1SegT[piNNo * 2].ml1Val[piKind] = sz1SegT[piNNo].ml1Val[piKind]; sz1SegT[piNNo * 2 + 1].ml1Val[piKind] = sz1SegT[piNNo].ml1Val[piKind]; sz1SegT[piNNo].ml1Val[piKind] = 0; } // 範囲の半分 int liHalf = (piNowE - piNowS + 1) / 2; // 両側の子へ fSegTUpRngMain(piKind, piUpS, piUpE, piUpVal, piNNo * 2, piNowS, piNowS + liHalf - 1); fSegTUpRngMain(piKind, piUpS, piUpE, piUpVal, piNNo * 2 + 1, piNowE - liHalf + 1, piNowE); return 0; } int fSegTUpRng( int piKind // 種類 , int piUpS // 更新範囲 - 開始 0~ , int piUpE // 更新範囲 - 終了 0~ , int piUpVal // 更新値 ) { return fSegTUpRngMain(piKind, piUpS, piUpE, piUpVal, 1, 0, siSCNo - 1); } // セグメントツリー - 1データ取得 int fSegTGetOne( int piKind // 種類 , int piGNo // 取得位置 0~ ) { int liRet; // 対象位置 piGNo += siSCNo; // 初期値 liRet = (int)sz1SegT[piGNo].ml1Val[piKind]; // 親に値があれば更新 while (piGNo > 1) { piGNo /= 2; // 上位へ if (sz1SegT[piGNo].ml1Val[piKind] != 0) { // 更新済み liRet = (int)sz1SegT[piGNo].ml1Val[piKind]; } } return liRet; } // 実行メイン int fMain( ) { int i; char lc1Buf[1024]; // 車両数・クエリ数 - 取得 int liCCnt, liQCnt; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d", &liCCnt, &liQCnt); // セグメントツリー - 子の開始位置 - セット fSegTSetCSNo(liCCnt); // かっこよさ - 取得 for (i = 0; i < liCCnt; i++) { fscanf(szpFpI, "%lld", &sz1SegT[siSCNo + i].ml1Val[D_SEGT_SUM]); } fgets(lc1Buf, sizeof(lc1Buf), szpFpI); // セグメントツリー - 親データセット fSegTSetPData(); // 最小車両・最大車両 - 初期化 for (i = 0; i < liCCnt; i++) { sz1SegT[siSCNo + i].ml1Val[D_SEGT_MIN] = i + 1; sz1SegT[siSCNo + i].ml1Val[D_SEGT_MAX] = i + 1; } // クエリ - 取得 for (i = 0; i < liQCnt; i++) { int liOpe, liCNo; fgets(lc1Buf, sizeof(lc1Buf), szpFpI); sscanf(lc1Buf, "%d%d", &liOpe, &liCNo); // クエリ - 実行 int liMin, liMax; switch (liOpe) { case 1: // 車両結合 liMax = fSegTGetOne(D_SEGT_MAX, liCNo - 1); if (liMax <= liCNo) { liMin = fSegTGetOne(D_SEGT_MIN, liCNo - 1); liMax = fSegTGetOne(D_SEGT_MAX, liCNo); fSegTUpRng(D_SEGT_MIN, liMin - 1, liMax - 1, liMin); fSegTUpRng(D_SEGT_MAX, liMin - 1, liMax - 1, liMax); } break; case 2: // 車両分割 liMax = fSegTGetOne(D_SEGT_MAX, liCNo - 1); if (liMax > liCNo) { liMin = fSegTGetOne(D_SEGT_MIN, liCNo - 1); fSegTUpRng(D_SEGT_MAX, liMin - 1, liCNo - 1, liCNo); liMax = fSegTGetOne(D_SEGT_MAX, liCNo); fSegTUpRng(D_SEGT_MIN, liCNo, liMax - 1, liCNo + 1); } break; case 3: // かっこよさ増加 fSegTUpOne(liCNo - 1); break; case 4: // かっこよさ表示 liMin = fSegTGetOne(D_SEGT_MIN, liCNo - 1); liMax = fSegTGetOne(D_SEGT_MAX, liCNo - 1); long long llSum = fSegTGetRng(liMin - 1, liMax - 1); sprintf(lc1Buf, "%lld\n", llSum); fOut(lc1Buf); break; } } return 0; } // 1回実行 int fOne( ) { int liRet; char lc1Buf[1024]; // データ - 初期化 memset(sz1SegT, 0, sizeof(sz1SegT)); // セグメントツリー // 入力 - セット #ifdef D_TEST sprintf(lc1Buf, ".\\Test\\T%d.txt", siTNo); szpFpI = fopen(lc1Buf, "r"); sprintf(lc1Buf, ".\\Test\\A%d.txt", siTNo); szpFpA = fopen(lc1Buf, "r"); siRes = 0; #else szpFpI = stdin; #endif // 実行メイン liRet = fMain(); // 残データ有無 #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", siTNo); } else { printf("NG %d\n", siTNo); } #endif return 0; } // プログラム開始 int main() { #ifdef D_TEST int i; for (i = D_TEST_SNO; i <= D_TEST_ENO; i++) { siTNo = i; fOne(); } #else fOne(); #endif return 0; }