AtCoder Regular Contest 107 B - Quadruple

题目链接

题目翻译：

给你两个整数N和K，求有多少种组合方式 (a,b,c,d)，使其同时满足下面两个条件：

• 1 ≤ a,b,c,d ≤ N
• a + b - c - d = K

解题思路：

首先来分析一下a + b - c - d = K：
a + b - c - d = K => a + b - (c + d) = K
把a+b，c+d看做两个整体A、B，即A - B = K（A = B + K，B = A - K）
可以发现K的正负，对结果并没有影响，所以我们直接对K取绝对值，方便后面的计算。
再来分析一下A和B的取值范围，因为K现在是正的，所以A肯定大于等于B，B的最小值是2，所以A的最小值是2+K，A的最大值是2×N，所以B的最大值2×N-K。即A∈[2+K,2×N]，B∈[2,2×N-K]。
再往细了分析：
在确定A的情况下，怎么计算a + b = A有多少种组合方式？
同样，我们先来确定a和b的取值范围。
可知a的最小值受b的最大值和A的影响，同时又受1≤a≤N的影响，所以a的最小值mina是max(A-N,1)。a的最大值受b的最小值和A的影响，同时又受1≤a≤N的影响，所以a的最大值maxa是min(A-1,N)。而a，b是可以交换的，所以a的取值范围就是b的取值范围。因此在确定A的情况下，a,b共有maxa-mina+1种组合方式。
同理可以知道在该情况下，c，d共有min(B-1,N)-max(B-N,1)+1种组合方式。
所以共有 (min(A-1,N)-max(A-N,1)+1)×(min(B-1,N)-max(B-N,1)+1) 种可能
然后我们遍历A的所有可能取值，每次遍历累加可能的组合数，就可以得到最终的结果。

我的代码：

#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<cstdio>
#include<cmath>
#define inf 0x3f3f3f3f
using namespace std;
int main(){
//	freopen("1.txt","r",stdin);
	long long n,k,a,b,maxa,mina,maxb,minb;
	long long ans=0;
	cin>>n>>k;
	k=abs(k);
	for(int i=2+k;i<=2*n;i++){//
		a=i,b=i-k;
		maxa=(long long)min(n,a-1);
		mina = (long long)max(1LL,a-n);
		maxb=(long long)min(n,b-1);
		minb = (long long)max(1LL,b-n);
		ans+=(maxa-mina+1)*(maxb-minb+1);
	}
	cout<<ans<<endl;
	return 0;
}

总结：

又是long long，又没有估计数据的范围