結果

問題 No.498 ワープクリスタル (給料日編)
ユーザー myantamyanta
提出日時 2017-06-30 09:24:55
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 468 ms / 2,000 ms
コード長 1,729 bytes
コンパイル時間 977 ms
コンパイル使用メモリ 54,328 KB
実行使用メモリ 101,072 KB
最終ジャッジ日時 2024-04-15 06:42:06
合計ジャッジ時間 6,067 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 1 ms
6,944 KB
testcase_03 AC 1 ms
6,944 KB
testcase_04 AC 242 ms
64,764 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 229 ms
82,296 KB
testcase_07 AC 232 ms
64,284 KB
testcase_08 AC 292 ms
90,432 KB
testcase_09 AC 365 ms
92,392 KB
testcase_10 AC 1 ms
6,944 KB
testcase_11 AC 34 ms
17,848 KB
testcase_12 AC 7 ms
7,424 KB
testcase_13 AC 27 ms
16,712 KB
testcase_14 AC 10 ms
9,528 KB
testcase_15 AC 13 ms
12,256 KB
testcase_16 AC 6 ms
6,944 KB
testcase_17 AC 15 ms
13,196 KB
testcase_18 AC 419 ms
95,072 KB
testcase_19 AC 389 ms
86,984 KB
testcase_20 AC 423 ms
98,348 KB
testcase_21 AC 468 ms
92,240 KB
testcase_22 AC 465 ms
101,072 KB
testcase_23 AC 344 ms
91,192 KB
testcase_24 AC 263 ms
82,088 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:82:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   82 |                         scanf("%d%d%d", &c[i].x, &c[i].y, &c[i].n);
      |                         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include<cstdio>
#include<vector>


using namespace std;
using vi=vector<int>;
using vvi=vector<vi>;
using vvvi=vector<vvi>;
using ll=long long;

#define MOD 1000000007

struct c_t
{
	int x, y, n;
};


ll mod_pow(ll a, ll b)
{
	ll r=1;

	for(;b>0;b>>=1)
	{
		if(b&1) r=(r*a)%MOD;
		a=(a*a)%MOD;
	}
	return r;
}


ll mod_inv(ll a)
{
	return mod_pow(a, MOD-2);
}


ll mod_fact(int n)
{
	ll r=1;

	for(;n>0;n--)
	{
		r=(r*n)%MOD;
	}
	return r;
}


ll calc(vi&ans)
{
	int k=ans.size();
	int n=0;
	ll u=1, l=1, r;

	for(int i=0;i<k;i++)
	{
		l=(l*mod_fact(ans[i]))%MOD;
		n+=ans[i];
	}
	u=mod_fact(n);

	r=(u*mod_inv(l))%MOD;
//	printf("u=%lld  l=%lld  r=%lld\n", u, l, r);

	return r;
}


int main(void)
{
	int gx, gy, k;
	int offset_x=750000;

	while(scanf("%d%d%d", &gx, &gy, &k)==3)
	{
		vector<c_t> c(k);
		int max_x=0, min_x=0;

		for(int i=0;i<k;i++)
		{
			scanf("%d%d%d", &c[i].x, &c[i].y, &c[i].n);
			max_x=max(max_x, max_x+c[i].x*c[i].n);
			min_x=min(min_x, min_x+c[i].x*c[i].n);
			offset_x=-min_x;
		}
		vvvi dp(max_x-min_x+1), w;
		dp[offset_x].resize(1);
		dp[offset_x][0].resize(k);
		for(int i=0;i<k;i++)
		{
			w=dp;
			for(int x=0;x<w.size();x++)
			{
				if(!w[x].size()) continue;
				for(int j=0;j<w[x].size();j++)
				{
					for(int l=1;l<=c[i].n;l++)
					{
						vi t=w[x][j];
						t[i]=l;
						int nx=x+c[i].x*l;
						dp[nx].push_back(t);
					}
				}
			}
		}

		vvi&dpe=dp[gx+offset_x];
		ll ans=0;
		for(int i=dpe.size()-1;i>=0;i--)
		{
			int y=0;
			for(int j=0;j<k;j++)
			{
				y+=dpe[i][j]*c[j].y;
			}
			if(y==gy)
			{
				ans=(ans+calc(dpe[i]))%MOD;
/*
				for(int j=0;j<k;j++)
				{
					printf("%d ", dpe[i][j]);
				}
				printf("\n");
*/
			}
		}
		printf("%lld\n", ans);
	}
	return 0;
}
0