Joseph‘s Problem（UVA 1363）

数论学习笔记

Joseph’s Problem（UVA 1363）

题意：

输入正整数n和k（1<=n,k<=1e9）,计算 ${\sum }_{i=1}^n$k mod i。

思路：

先把k%i == k的部分处理掉，然后就变成了只剩下对k>n的情况。
假设k/i的整数部分为p，因为k/(i+1)和k/i的整数部分差别不大，当k/(i+1)的整数部分也为p的时候:

k%(i+1) = k-(i+1)p = k-ip-p = k%i - p

也就是说，当p相同时，对于连续的区间i,i+1,i+2,…,j,k除以他们的余数是一个等差数列。这样就可以枚举i的值，将对应的等差数列的和累加到答案中。

代码：

#include<algorithm>
#include<stdio.h>
#include<iostream>
using namespace std;
typedef long long LL;

int main(){
	int n,k;
	while(scanf("%d%d",&n,&k)==2){
		LL ans = 0;
		if(n>=k){//清掉k%i==k的值 
			ans = (LL)(n-k)*k;
			n = k-1;
		}
		
		for(int i=1,j;i<=n;i=j+1){
			j=min(k/(k/i),n);
			ans+=(LL)(k%i+k%j)*(LL)(j-i+1)/2;
		}
		printf("%lld\n",ans);
	}	
} 

思路来源于刘汝佳紫书P338。