BNUOJ 24251 Counting Pair 配对计数 【数论】

题目链接：

　　http://acm.bnu.edu.cn/v3/problem_show.php?pid=24251

---

数论：

　　给定1~N， 1~M的编号作为男女的编号

　　给定Q个查询，每个查询给出一个数字qnum，求出所有男女编号相加等于qnum的组合的情况

　　最大的组合情况就是N/M中最小的那个数

　　找规律发现在2~N+M这些数的两侧从1~min(N, M)-1进行排列，其他的位置填上min(N, M)即可

---

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
#define clr(c) memset(c, 0, sizeof(c));
const int MAXL = 200000+5;
int ans[MAXL];
int t, n, m, q, qnum;

void pro(){
    int maxnum = min(n, m);
    int sum = n+m;
    for(int i = 1; i < maxnum; i++){
        ans[2+i-1] = i;
        ans[sum-i+1] = i;
    }
    for(int i = 2+maxnum-1; i <= sum-maxnum+1; i++){
        ans[i] = maxnum;
    }
}

int result(){
    if(qnum > n+m || qnum == 0 || qnum == 1) return 0;
    else{
        return ans[qnum];
    }
}

int main(){
    scanf("%d", &t);
    for(int Case = 1; Case <= t; Case++){
        clr(ans);
        scanf("%d%d", &n, &m);
        scanf("%d", &q);
        printf("Case #%d:
", Case);
        pro();
        while(q--){
            scanf("%d", &qnum);
            printf("%d
", result());
        }
    }

    return 0;
}