选择客栈X无线通讯网 二合一

啥都不说了直接上码

唯一一点就是第一份码里的sum和f，sum[i][j]就是到i点的时候j颜色的数量，f[i]就是距离i最近的左边的客栈

然后就是第二个中，s-1是可以消去权值的边，p-1是生成树的边，p-1-（s-1）就等于p-s,就是最终循环跳出的边

#include<iostream>
using namespace std;
const int MAXN=200005;
int qian[MAXN][51]; //qian[i][j]为到i时j的颜色数量 
int a[MAXN];
int c[MAXN];
int f[MAXN];
int n,k,p;
int ans;
int main()
{
cin>>n>>k>>p;
for(int i=1;i<=n;i++)
{
cin>>c[i]>>a[i];
for(int j=0;j<k;j++)
qian[i][j]=qian[i-1][j];
qian[i][c[i]]++;
if(a[i]<=p)
f[i]=i;
else f[i]=f[i-1];
}
for(int i=1;i<=n;i++)
{
ans+=qian[f[i]][c[i]];
if(f[i]==i)ans--;
}
cout<<ans;
return 0;
}

/

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
using namespace std;
int s,p;
struct line
{
int from,to;
long long w;
}edge[250005];
struct point
{
int x,y;
}pos[501];
int pre[505];
int rank[505];
void init()
{
for(int i=1;i<=p;i++)
pre[i]=i;
}
int find(int x)
{
if(pre[x]==x)return x;
else return pre[x]=find(pre[x]);
}
void unite(int x,int y)
{
x=find(x);
y=find(y);
if(rank[x]>rank[y])pre[y]=x;
else 
{
pre[x]=y;
if(rank[x]==rank[y])rank[y]++;
}
}
bool cmp(line a,line b)
{
return a.w<b.w;
}
long long ans;
int k;
int main()
{
cin>>s>>p;
for(int i=1;i<=p;i++)
cin>>pos[i].x>>pos[i].y;
int sum=0;
for(int i=1;i<=p;i++)
for(int j=i+1;j<=p;j++)
{
edge[sum].from=i;
edge[sum].to=j;
edge[sum].w=abs(pos[i].x-pos[j].x)*abs(pos[i].x-pos[j].x)+abs(pos[i].y-pos[j].y)*abs(pos[i].y-pos[j].y);
sum++;
}
init();
sort(edge,edge+sum,cmp);
for(int i=0;i<sum;i++)
{
if(find(edge[i].from)!=find(edge[i].to))
{
unite(edge[i].from,edge[i].to);
if(edge[i].w>ans)ans=edge[i].w;
k++;
}
if(k==p-s)break;
}
printf("%.2f",sqrt(ans));
return 0;
}