第 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;
}