#31329 (C++11) No.78 クジ付きアイスバー

• 問題: No.78 クジ付きアイスバー
• ユーザー: ty70
• 提出日時: 2015-06-08 02:13:19
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 2 ms / 5,000 ms
• コード長: 2,031 bytes
• コンパイル時間: 720 ms
• コンパイル使用メモリ: 90,116 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-11-08 00:31:06
• 合計ジャッジ時間: 1,804 ms
• ジャッジサーバーID （参考情報）: judge5 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	2 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	2 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	2 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	2 ms; 5,248 KB
testcase_11	AC	2 ms; 5,248 KB
testcase_12	AC	2 ms; 5,248 KB
testcase_13	AC	2 ms; 5,248 KB
testcase_14	AC	2 ms; 5,248 KB
testcase_15	AC	2 ms; 5,248 KB
testcase_16	AC	2 ms; 5,248 KB
testcase_17	AC	1 ms; 5,248 KB
testcase_18	AC	2 ms; 5,248 KB
testcase_19	AC	2 ms; 5,248 KB
testcase_20	AC	2 ms; 5,248 KB
testcase_21	AC	2 ms; 5,248 KB
testcase_22	AC	2 ms; 5,248 KB
testcase_23	AC	2 ms; 5,248 KB
testcase_24	AC	2 ms; 5,248 KB
testcase_25	AC	2 ms; 5,248 KB
testcase_26	AC	2 ms; 5,248 KB
testcase_27	AC	2 ms; 5,248 KB
testcase_28	AC	2 ms; 5,248 KB
testcase_29	AC	2 ms; 5,248 KB
testcase_30	AC	2 ms; 5,248 KB
testcase_31	AC	2 ms; 5,248 KB
testcase_32	AC	2 ms; 5,248 KB
testcase_33	AC	2 ms; 5,248 KB
testcase_34	AC	2 ms; 5,248 KB
testcase_35	AC	1 ms; 5,248 KB
testcase_36	AC	1 ms; 5,248 KB
testcase_37	AC	2 ms; 5,248 KB
testcase_38	AC	2 ms; 5,248 KB

ソースコード

#include <iostream>
#include <vector>
#include <string>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <algorithm>	// require sort next_permutation count __gcd reverse etc.
#include <cstdlib>	// require abs exit atof atoi 
#include <cstdio>		// require scanf printf
#include <functional>
#include <numeric>	// require accumulate
#include <cmath>		// require fabs
#include <climits>
#include <limits>
#include <cfloat>
#include <iomanip>	// require setw
#include <sstream>	// require stringstream 
#include <cstring>	// require memset
#include <cctype>		// require tolower, toupper
#include <fstream>	// require freopen
#include <ctime>		// require srand
#define rep(i,n) for(int i=0;i<(n);i++)
#define ALL(A) A.begin(), A.end()

using namespace std;

typedef long long ll;
typedef pair<int, int> P;

// シミュレーションで求める
string smulation (int & lucky, int & eaten, int & buy, string S, int N, int K ){
	string res = "";
	for (int i = 0; i < N;  i++ ){
		int curr = (S[i] - '0' );	// 現在のアイスバーの判定
		if (lucky > 0 ){
			res += 'x';//cerr << 'x'; アイスバーを買わなくて良い
			lucky--;
		}else{
			res += 'o';//cerr << 'o'; アイスバーを買う必要がある
			buy++;	
		} // end if
		eaten++;
		if (eaten == K ){
			return res;
		} // end if
		lucky += curr;
	} // end for

	return res;
}

int solve (int N, int K, string S ){
	int lucky = 0, eaten = 0, buy = 0;
	int res = 0;
	string t = smulation (lucky, eaten, buy, S, N, K );
	res += buy;
//	cerr << t << " buy: " << buy << " lucky: " << lucky << endl;
	if (N < K ){
		rep (i, N )
			if (t[i] == 'o' && lucky ) t[i] = 'x', lucky--;
//		cerr << t << endl;
		res += (int)count (ALL (t ), 'o' )*(K/N-1);
		if(K % N != 0 ){
			res += (int)count (t.begin(), t.begin()+(K%N), 'o' );
		} // end if
	} // end if

	return res;
}

int main()
{
	ios_base::sync_with_stdio(0);
	int N, K; cin >> N >> K;
	string S; cin >> S;

	int res = solve (N, K, S );
	cout << res << endl;

	return 0;
}