【JZOJ 4312】【NOIP2015模拟11.4】格点统计

问题描述

输入

输出

样例输入
输入1：
3
输入2：
4
样例输出
输出1：
5

输出2：
8
算法讨论
数学方法。我们将每种组合列一下：
1,1 2,1 3,1 4,1 5,1 6,1 7,1 8,1 9，1
2,1 2,2 3,2 4,2
3,1 2,3 3,3
4,1 2,4
5,1
6,1
7,1
8,1
9,1
我们会发现它是关于对角线对称的（易证明），那这样就很好求了。

#include <cstdio>
#include <math.h>
#include <cmath>
using namespace std;
#define ll long long
#define mo 998244353
ll k,s=0,s1=0;

int main()
{
    freopen("count.in","r",stdin);
    freopen("count.out","w",stdout);
    scanf("%lld",&k);
    for (ll i=1;i<=sqrt(k);i++)
    {
        s1++;
        s=(s+(k / i)-i) % mo;
    }
    s=(s*2+s1) % mo;
    printf("%d",s);
    fclose(stdin); fclose(stdout);
    return 0;
}