POJ2010 Moo University - Financial Aid —— 二分答案+巧妙的判断

二分答案，但是这里的判断不同于一般的二分。

分别将cows和cowc按照s和c排序。

我的代码里check函数的返回值含义如下：

-1 直接输出，不可能满足条件了

0 这种方案是满足条件的，可以把中位数调大试试

1 这种方案是不满足的，但是把中位数调大就有可能满足了（否则不可能满足）

2 这种方案是不满足的，但是把中位数调小就有可能满足了（否则不可能满足）

还是不会用cpp，用了switch语句却没写break，导致纠结了好长时间。

终于给家里台式机装上Ubuntu了，才发现NOI Linux就是害人用的啊。

 

参考代码

# include <cstdio>
# include <iostream>
# include <cstring>
# include <algorithm>
using namespace std;
const int maxN=100005;
struct Node{ int c,s,id; };
int n,m,tot;
Node cows[maxN],cowc[maxN];

bool cmps(Node a,Node b) { return a.s<b.s; }
bool cmpc(Node a,Node b) { return a.c<b.c; }

int check(int limit)
{
	int left=0,right=0,all=cows[limit].c;
	for (int i=0;i<n;i++){
		if ((cowc[i].id<limit)&&(all+cowc[i].c<=tot)&&(left<m/2)){
			all+=cowc[i].c;left++;
		} else if ((cowc[i].id>limit)&&(all+cowc[i].c<=tot)&&(right<m/2)){
			all+=cowc[i].c;right++;
		}
	}
	if ((left<m/2)&&(right<m/2)) return -1;
	else if (left<m/2) return 1;
	else if (right<m/2) return 2;
	else return 0;
}

int main()
{
	scanf("%d%d%d",&m,&n,&tot);
	for (int i=0;i<n;i++) scanf("%d%d",&cows[i].s,&cows[i].c);
	sort(cows,cows+n,cmps);
	for (int i=0;i<n;i++) cows[i].id=i;
	memcpy(cowc,cows,sizeof(cows));
	sort(cowc,cowc+n,cmpc);
	int l=1,r=n,ans=-1;
	while (l<=r){
		int mid=(l+r)>>1;
		switch (check(mid)){
			case -1: { printf("-1\n"); return 0; }
			case  0: { ans=cows[mid].s; l=mid+1; break; }
			case  1: { l=mid+1; break;}
			case  2: { r=mid-1; }
		}
	}
	if (ans>=0) printf("%d\n",ans); else printf("-1\n");
	return 0;
}