小易是一个数论爱好者,并且对于一个数的奇数约数十分感兴趣。一天小易遇到这样一个问题: 定义函数f(x)为x最大的奇数约数,x为正整数。 例如:f(44) = 11.
现在给出一个N,需要求出 f(1) + f(2) + f(3)…….f(N)
例如: N = 7
f(1) + f(2) + f(3) + f(4) + f(5) + f(6) + f(7) = 1 + 1 + 3 + 1 + 5 + 3 + 7 = 21
小易计算这个问题遇到了困难,需要你来设计一个算法帮助他。
输入描述:
输入一个整数N (1 ≤ N ≤ 1000000000)
输出描述:
输出一个整数,即为f(1) + f(2) + f(3)…….f(N)
分析:
1.看到 1 ≤ N ≤ 1000000000 N > 10^8 说明算法的时间复杂度要低于 O(n),那基本可以确定算法为O(logn)
2. 先利用递归的方法,层层递进,时间复杂度O(n)
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
// int[] memo = new int[n+1]; // memo[i] 表示 i 的全记录。 memo[i] = memo[i-1]+fun[i]
int fun1 = 1; //表示当前的奇约数
int sum = 0;
if(n % 2 == 1){
int x = n+1/2+1;
sum = x * x;
}
int i = 2;
for(i = 2; i <= n; i =i+2){ //单独列出来,还是不对, i = i+2 依然还是O(n) 的算法,
sum += getAppNums(i); //题目要求的是 O(logn)的算法,因此要考虑到除2的
}
System.out.println(sum);
}
public static int getAppNums(int n) {
if(n % 2 == 1)
return n;
while( n % 2 != 1){
n = n /2;
}
return n;
}
上面的算法思想是,奇数最大奇约数就是本身,因此只需要迭代添加偶数的最大奇约数是循环除/2得到的奇数。
尽管迭代是 i = i+2 ,但时间复杂度还是O(n),要达到O(logn)一般都有 n / 2 这种类似于二分搜索。所以,继续往这上面思考。
考虑到偶数都是除于2,那循环除于2,如
1 2 3 4 5 6 7 8 其中奇数 1 3 5 7
偶数全部除于2 得到 1 2 3 4 其中奇数为 1 3
继续除于2 得到 1 2 奇数为 1
。。
所以得到代码:
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
long n = scanner.nextLong();
long sum = 0;
for(long i = n; i > 0; i = i / 2){
sum += ((i+1)/2) * ((i+1)/2); // 奇数等差数列
}
System.out.println(sum);
}