【cf 906 D】

Doremy's Connecting Plan

Doremy生活在一个由编号从1 到n的n个城市组成的国家，第二个城市居住着ai人。它可以被建模为具有n节点的无向图。最初，图形中没有边。现在Doremy想要使图连通。为此，她可以在i和j之间添加一条边，如果：

其中S是当前在 i 或 j 的同一连接分量中的所有节点的集合，并且 c 是给定的常数。Doremy能把图连接起来吗？

反证法证 ai，aj中至少有一个与a1相连；

如果 ai 和 aj 都不能与a1相连，那么：

ai+aj>=i*j*c;

ai+a1<i*c;

aj+a1<j*c;

其中 i，j 大于等于2；

得 (i+j)*c>i*j*c+2*a1>=i*j*c;

即 i+j>i*j，i<j/(j-1)<=2;

所以，i<2，与 i大于等于2矛盾；

如果该点可以连通，那么该点一定和a1相连；

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<deque>
using namespace std;
typedef long long LL; 
const int N=2e5+10;
LL a[N];
pair<LL,int> pii[N];
int main()
{
	int T;
	cin>>T;
	while(T--)
	{
		int n,c;
		cin>>n>>c;
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
			if(i!=1) pii[i].first=(LL)i*c-a[i];
			pii[i].second=i;
		}
		//从小到大排序，sum<i*c-a[i]的最小值，那么sum也一定小于后面的点，
		//即找不到其他点可以与该图联通； 
		sort(pii+2,pii+n+1);
		LL sum=a[1];
		bool flag=1;
		for(int i=2;i<=n;i++)
		{
			LL t=pii[i].first;
			int id=pii[i].second;  
			if(sum<t)
			{
				flag=0;
				break;
			}
			sum+=a[id];
		}
		if(flag) cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
	}
	return 0;
 }