P3029 [USACO11NOV] Cow Lineup S
题目描述
问题描述
农民约翰雇一个专业摄影师给他的部分牛拍照。由于约翰的牛有好多品种,他喜欢他的照片包含每个品种的至少一头牛。
约翰的牛都站在一条沿线的不同地方, 每一头牛由一个整数位置 X i X_i Xi 以及整数品种编号 I D i ID_i IDi 表示。
约翰想拍一张照片,这照片由沿线的奶牛的连续范围组成。照片的成本与规模相当,这就意味着,在一系列照片中的最大和最小 X X X 坐标的差距决定了照片的成本。
请帮助约翰计算最小的照片成本,这些照片中有每个不同的品种的至少一头牛,没有两头牛愿意站在同一个地点的。
输入格式
第 1 1 1 行:牛的数量 N N N;
输出格式
输出共一行,包含每个不同品种 I D \rm ID ID 的照片的最低成本。
输入输出样例 #1
输入 #1
6
25 7
26 1
15 1
22 3
20 1
30 1
输出 #1
4
代码内容
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=5e4+10;
struct node
{
ll x,id;
}cow[N],que[N];
map<ll,ll> m;
bool cmp(node p,node q)
{
return p.x<q.x;
}
int main()
{
ll n;
cin>>n;
ll sum=0;
for(ll i=1;i<=n;i++)
{
cin>>cow[i].x>>cow[i].id;
m[cow[i].id]++;
if(m[cow[i].id]==1) sum++;//计算奶牛种数
}
sort(cow+1,cow+n+1,cmp);
m.clear();
ll l=1,r=0;// l左端点 r右端点
ll ans=1e18,cnt=0;//cnt:照片中的奶牛种数
for(ll i=1;i<=n;i++)
{
if(!m[cow[i].id]) m[cow[i].id]=1,cnt++;//如果cow[i].id不在mp中
else m[cow[i].id]++;
que[++r]={cow[i].x,cow[i].id};
while(m[que[l].id]>1) m[que[l++].id]--;//l++左端点右移一位(去重)
//照片中奶牛种数等于一共的奶牛种数
if(cnt==sum) ans=min(ans,que[r].x-que[l].x);
}
cout<<ans<<endl;
return 0;
}