洛谷刷题日记-P1011 [NOIP1998 提高组] 车站

1.题目描述
火车从始发站（称为第1站）开出，在始发站上车的人数为a，然后到达第2站，在第2站有人上、下车，但上、下车的人数相同，因此在第2站开出时（即在到达第3站之前）车上的人数保持为
a人。从第3站起（包括第3站）上、下车的人数有一定规律：上车的人数都是前两站上车人数之和，而下车人数等于上一站上车人数，一直到终点站的前一站（第n−1站），都满足此规律。现给出的条件是：共有n个车站，始发站上车的人数为a，最后一站下车的人数是m（全部下车）。试问x站开出时车上的人数是多少？
2.输入格式
输入只有一行四个整数，分别表示始发站上车人数a，车站数n，终点站下车人数m和所求的站点编号x。
3.输出格式
输出一行一个整数表示答案：从x站开出时车上的人数。
4.样例
输入#1

5 7 32 4

输出#1

13

5.说明/提示
对于全部的测试点，保证1≤a≤20，1≤x≤n≤20，1≤m≤2×10000 。
NOIP1998 提高组 第一题
6.AC CODE

#include<bits/stdc++.h>
using namespace std;
int a, n, m, x, u=1, z, y;
int main()
{
    scanf("%d %d %d %d", &a, &n, &m, &x); 
    if(n <= 5) {
        if(n == 2||n == 3)
            printf("%d", a);
        else if(n == 4) {
            if(x == 1 || x == 2) printf("%d", a);
            else if(x == 3) printf("%d", a * 2);
        }
        else if(n == 5) {
        if(x == 1 || x == 2) printf("%d", a);
            else if(x == 3) printf("%d", a * 2);
            else if(x == 4) 
                printf("%d", (m - a * 3) / 2 + a * 2);
        }
    }
    else {
        int p = 1, q = 0, k = 0, sum1 = 0;
        for(int i = 1; i <= n - 5; i++) { 	
            k = p + q;
            sum1 += k;
            p = q;
            q = k;
        }
        int s1 = sum1 + 2;
        int e = 0, t = 1, g = 0,sum2 = 0;
        for(int i = 1; i <= n - 5; i++) {
            g = e + t;
            sum2 += g;
            e = t;
            t = g;
        }
        int s2 = sum2 + 1;
        int S = (m - s1 * a) / s2;
        q = k = e = g = sum1 = sum2 = 0;
        p = t = 1;
        if(x <= 5) {
            if(x == 1 || x == 2) printf("%d", a);
            else if(x == 3)  printf("%d", a * 2);
            else if(x == 4) printf("%d", S + a * 2);
            else printf("%d", S * 2 + a * 3);
        }
        else {
            for(int i = 1; i <= x - 4; i++) {
                k = p + q;
                sum1 += k;
                p = q;
                q = k;
            }
            sum1 += 2;
            for(int i = 1; i <= x - 4; i++) {
                g = e + t;
                sum2 += g;
                e = t;
                t = g;
            }
            sum2 += 1;
            printf("%d", sum1 * a + sum2 * S);
        }
    }
    return 0;
}