codevs 1083 Cantor表

题目描述 Description

现代数学的著名证明之一是Georg Cantor证明了有理数是可枚举的。他是用下面这一张表来证明这一命题的： 1/1 1/2 1/3 1/4 1/5 … 2/1 2/2 2/3 2/4 … 3/1 3/2 3/3 … 4/1 4/2 … 5/1 … … 我们以Z字形给上表的每一项编号。第一项是1/1，然后是1/2，2/1，3/1，2/2，…

输入描述 Input Description

整数N（1≤N≤10000000）

输出描述 Output Description

表中的第N项

#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
#define MAX 100010
using namespace std;

int main() {
    //while(1) {
        int n, y, i, s = 0;
        cin >> n;n--;
        for(i = 1; i <= 1000000; i++) {
            s+=i;
            if(n < s)
                break;
        }
        i--;n++;
        y = (i % 2 == 0) ? (i + 1) * (i / 2) : i * (i / 2 + 1);
        i++;
        int j = (i % 2 == 0) ? (i + 1) * (i / 2) : i * (i / 2 + 1);
        if(i % 2 == 0) {
            int a = n - y;
            int b = j - n + 1;
            cout << a << "/" << b;
        } else {
            int b = n - y;
            int a = j - n + 1;
            cout << a << "/" << b;
        }
    //}
    return 0;
}