結果

問題 No.1201 お菓子配り-4
ユーザー kotatsugame
提出日時 2020-11-27 10:20:35
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 774 bytes
コンパイル時間 593 ms
コンパイル使用メモリ 42,564 KB
実行使用メモリ 9,636 KB
最終ジャッジ日時 2024-07-23 21:26:51
合計ジャッジ時間 33,219 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 35 TLE * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:34:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   34 | main()
      | ^~~~

ソースコード

diff # 

#include<cstdio>
#include<algorithm>
using namespace std;
long long floor_sum(long long N,long long M,long long A,long long B)
//Sum[floor((A*i+B)/M),{i,0,N-1}]
{
	long long ans=0;
	while(true)
	{
		if(A>=M)
		{
			long long AM=A/M;
			ans+=N*(N-1)/2*AM;
			A-=AM*M;
		}
		if(B>=M)
		{
			long long BM=B/M;
			ans+=N*BM;
			B-=BM*M;
		}
		long long Ym=(A*N+B)/M,Xm=Ym*M-B;
		if(Ym==0)break;
		long long TX=(Xm+A-1)/A;
		ans+=(N-TX)*Ym;
		N=Ym;
		B=TX*A-Xm;
		swap(A,M);
	}
	return ans;
}
int N,M;
int A[2000],B[2000];
main()
{
	scanf("%d%d",&N,&M);
	for(int i=0;i<N;i++)scanf("%d",&A[i]);
	for(int i=0;i<M;i++)scanf("%d",&B[i]);
	long ans=0;
	for(int i=0;i<N;i++)for(int j=0;j<M;j++)
	{
		ans+=2*floor_sum(B[j],B[j],A[i],A[i]);
		ans%=(int)1e9+7;
	}
	printf("%ld\n",ans);
}
0