题目链接:
D - Staircase Sequenceshttps://atcoder.jp/contests/abc190/tasks/abc190_d
思路:
要求公差为1的数列的数量,肯定要用到等差数列求和公式,假设是[a, b]部分求和,和为n,可以得到(a+b)(b-a+1) = 2n,所以就转化为求2n的两个因子的数量就可以了。首先我们观察到a+b和b-a+1肯定都是正的,所以排除两个负因子的情况。又因为(a+b)+(b-a+1) = 2b+1,是奇数,所以两个因子必定一奇一偶。
而又因为每对因子对应一组a和b的取值,所以到这里,问题就转化为求2n的一奇一偶的一对正因子的数量问题,所以我们枚举一个正因子,让另一个因子和它相加为奇数即可,根据对称性,只需要枚举到 根号2n ,然后结果乘以二就是答案啦。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <cmath>
using namespace std;
#define int long long
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
int n;
cin >> n;
int ans = 0;
n *= 2;
for(int i=1; i*i<=n; i++){
if(n % i == 0 && (i+n/i)%2 == 1){
ans++;
}
}
cout << ans*2 << endl;
}