結果

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

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#include<cstdio>
#include<algorithm>
using namespace std;
unsigned long floor_sum(unsigned int N,unsigned int M,unsigned int A,unsigned int B)
//Sum[floor((A*i+B)/M),{i,0,N-1}]
{
	unsigned long ans=0;
	if(B>=M)
	{
		unsigned int BM=B/M;
		ans+=(unsigned long)N*BM;
		B-=BM*M;
	}
	while(true)
	{
		unsigned int AM=A/M;
		ans+=(unsigned long)N*(N-1)/2*AM;
		A-=AM*M;
		if((unsigned long)A*N+B<M)break;
		unsigned long Ym=((unsigned long)A*N+B)/M,Xm=Ym*M-B;
		unsigned 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]);
	unsigned 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