結果

問題 No.1935 Water Simulation
コンテスト
ユーザー suzushi1i
提出日時 2022-05-14 01:49:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 954 bytes
コンパイル時間 900 ms
コンパイル使用メモリ 80,412 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-22 06:49:00
合計ジャッジ時間 1,867 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

//https://yukicoder.me/submissions/753613
#include<iostream>
#include<array>
#include<map>
using namespace std;
#define N 4
int main()
{
	array<int, N>V; for (int i = 0; i < N; cin >> V[i++]);
	long long X; cin >> X;

	auto next = [&](pair<array<int, N>, int>A)
	{
		int from = A.second, to = (from + 1) % N;
		const int move = min(A.first[from], V[to] - A.first[to]);
		A.first[from] -= move, A.first[to] += move;
		A.second = to;
		return A;
	};

	auto slv = [&](pair<array<int, N>, int>A, long long cnt, auto slv)
	{
		if (cnt == 0)return A;
		static map<pair<array<int, N>, int>, long long>memo;
		if (memo.count(A))
		{
			cnt %= memo[A] - cnt;
			while (cnt--)A = next(A);
			return A;
		}
		else
		{
			memo[A] = cnt;
			return slv(next(A), --cnt, slv);
		}
	};

	array<int, N>start = { 0 }; start[0] = V[0];
	auto ans = slv(make_pair(start, 0), X, slv);

	for (int i = 0; i < N; i++)cout << ans.first[i] << (i != N - 1 ? ' ' : '\n');
	return 0;
}
0