結果

問題 No.1201 お菓子配り-4
ユーザー kotatsugame
提出日時 2020-11-27 10:25:02
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 762 bytes
コンパイル時間 228 ms
コンパイル使用メモリ 40,576 KB
最終ジャッジ日時 2025-01-16 06:00:55
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 35 TLE * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:33:1: warning: ISO C++ forbids declaration of ‘main’ with no type [-Wreturn-type]
   33 | main()
      | ^~~~
main.cpp: In function ‘int main()’:
main.cpp:35:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   35 |         scanf("%d%d",&N,&M);
      |         ~~~~~^~~~~~~~~~~~~~
main.cpp:36:34: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   36 |         for(int i=0;i<N;i++)scanf("%d",&A[i]);
      |                             ~~~~~^~~~~~~~~~~~
main.cpp:37:34: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   37 |         for(int i=0;i<M;i++)scanf("%d",&B[i]);
      |                             ~~~~~^~~~~~~~~~~~

ソースコード

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;
	if(B>=M)
	{
		long long BM=B/M;
		ans+=N*BM;
		B-=BM*M;
	}
	while(true)
	{
		if(A>=M)
		{
			long long AM=A/M;
			ans+=N*(N-1)/2*AM;
			A-=AM*M;
		}
		if(A*N+B<M)return ans;
		long long Ym=(A*N+B)/M,Xm=Ym*M-B;
		long long TX=(Xm+A-1)/A;
		ans+=(N-TX)*Ym;
		N=Ym;
		B=TX*A-Xm;
		swap(A,M);
	}
}
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