Codeforces Round #728 (Div. 2)题解+补题

C. Great Graphs

一、题目链接

https://codeforces.ml/contest/1541/problem/C

二、题目大意

共有t组数据。存在一个有n个点的图，边可以为负数，但不存在负环。每个点对之间的边在一个方向上最多有一条。现在给一个长度为n的数组d，d[i]表示从1到i的最短距离。题目要求得出这个图能得到的最小边权和。

数据范围：1<=t<=10⁴，1<=n<=10⁵，$\sum n$<=2*10⁵，0<=d[i]<=10⁹。

三、题目分析

• 我们首先考虑一下边权都为非负数的情况，显然，此时将d数组从小到大排序，然后依次按d数组的差值作为相邻两个点的边，最后得到的边权和是最小的。
• 那其实有负边权的情况其实也是一样的。无论负边如何连接，由d确定的正边的最小值依然是第一步所得到的内容。所以只需要在原来没有负边权的图中添加负边来使答案更小即可。
• 为了不形成负环，所以如果i到j的正边权和为x，那么j到i的负边权和不能小于-x。如果每一对i，j都形成这样的负边，那么答案就最小了，为了使答案最小，显然要等于-x。
• 由第三条，我们可以假设，j到i的这一条边是由j到j-1，j-1到j-2等等这样相邻的边拼接出来的。于是我们就可以计算相邻两个点的连边对答案的贡献来得到答案。以i到i+1这条边为例，因为左边i个点和右边n-i个点，每一个点对都要形成第三条中提到的负边，所以这条边的使用次数就是$i\ast (n-i)$。最终这条边对答案的贡献就是$i\ast (n-i)\ast w$。

四、正解代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cmath>

using namespace std;
typedef long long ll;
const ll maxn=100010;
ll a[maxn];
int main()
{
	ll T,n;
	scanf("%lld",&T);
	while(T--)
	{
		scanf("%lld",&n);
		for(ll i=1;i<=n;i++)
			scanf("%lld",&a[i]);
		sort(a+1,a+1+n);
		ll ans=0;
		for(ll i=2;i<=n;i++)
		{
			ans+=a[i]-a[i-1];
			ans-=(i-1)*(n-i+1)*(a[i]-a[i-1]);
		}
		printf("%lld\n",ans);
	}
	
	return 0;
}