第 45 届国际大学生程序设计竞赛（ICPC）亚洲网上区域赛模拟赛 A题

第 45 届国际大学生程序设计竞赛（ICPC）亚洲网上区域赛模拟赛 A题

题目链接

1.题意

0 ≤ 𝑥 ≤ 𝑎, 0 ≤ 𝑦 ≤ 𝑏, 0 ≤ 𝑧 ≤ 𝑐, 0 ≤ 𝑘 ≤ 𝑑
输入a,b,c,d
问 𝑥 + 𝑦 + 𝑧 = 𝑘 有多少种组合

2.思路

𝑥 + 𝑦 = 𝑘 - 𝑧
先求出x+y的个数
再求出k-z的个数
相乘求和即可
参考

补图：k-z的个数

3.代码

#include "bits/stdc++.h"
typedef long long LL;
using namespace std;
const int N = 2e6+10;
LL a,b,c,d;
LL f1[N], f2[N];
int main(){
    cin >> a >> b >>c >>d;
    if(a > b) swap(a, b);
    for(int i = 0; i <= a+b; ++i){
        if(i <= a)f1[i] = i+1;
        else if(i <= b) f1[i] = a + 1;
        else f1[i] = a + b - i + 1;
    }
    int j = 0;
    for(int i = -c; i <= d; ++i ){
        if(i <= min(c,d)-c) f2[i] = ++j;
        else if( i <= max(c,d)-c) f2[i] =  min(c,d)+1;
        else f2[i] = --j;

    }
    LL sum = 0;
    for(int i = 0; i <= min(a+b, d); ++i)
        sum+=f1[i]*f2[i];
    cout <<sum<<endl;
    return 0;
}