A. Cow and Haybales
略
B. Cow and Friend
略
C. Cow and Message
略
D. Cow and Fields
略
E. Cow and Treats
注意到两边牛吃的范围不能相交,并且因为不能跨越,所以两边分别不能选喜好相同的牛。然后可以发现一个方案合法当且仅当分界点两侧的牛忽略其他奶牛可以吃饱(因为可以按要吃的草距离从大到小的顺序安排牛去吃)。
那么就很简单了。考虑枚举左侧牛吃到的最右的草 i i i,此时每种喜好的奶牛就独立了,可以分别做一个简单的贪心和计数,注意到若 i > 0 i>0 i>0,要求选择一只喜好 s i s_i si的牛吃掉这堆草。
枚举 i i i后可以暴力枚举所有牛,时间复杂度为 O ( n ( n + m ) ) \mathcal O(n(n+m)) O(n(n+m))。
#include <bits/stdc++.h>
#define MOD 1000000007
using namespace std;
typedef long long ll;
vector <int> vt[5005];
int num[5005],size1[5005],size2[5005];
int main() {
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&num[i]);
for(int i=1;i<=m;i++) {
int x,y;
scanf("%d%d",&x,&y);
vt[x].push_back(y);
}
for(int i=1;i<=n;i++) sort(vt[i].begin(),vt[i].end());
int ans1=0,ans2=0;
for(int i=0;i<=n;i++) {
memset(size1,0,sizeof(size1));
memset(size2,0,sizeof(size2));
for(int j=1;j<=i;j++) size1[num[j]]++;
for(int j=i+1;j<=n;j++) size2[num[j]]++;
int s1=0;
ll s2=1;
bool v=1;
for(int j=1;j<=n;j++) {
int r1=0,r2=0;
while (r1<vt[j].size()&&vt[j][r1]<=size1[j]) r1++;
while (r2<vt[j].size()&&vt[j][r2]<=size2[j]) r2++;
if (num[i]==j) {
int sz=0;
for(int k=0;k<vt[j].size();k++)
if (vt[j][k]==size1[j]) sz++;
if (!sz) {
v=0;
break;
}
r2-=(size1[j]<=size2[j]);
if (r2) {
s1+=2;
s2=s2*sz%MOD*r2%MOD;
}
else {
s1++;
s2=s2*sz%MOD;
}
}
else {
if (r1<r2) swap(r1,r2);
if (r1>1&&r2) {
s1+=2;
s2=s2*r2%MOD*(r1-1)%MOD;
}
else if (r1) {
s1++;
s2=s2*(r1+r2)%MOD;
}
}
}
if (!v) continue;
if (s1>ans1) {
ans1=s1;
ans2=s2;
}
else if (s1==ans1) ans2=(ans2+s2)%MOD;
}
printf("%d %d\n",ans1,ans2);
return 0;
}
F. Cow and Vacation
考虑把每条边取中点拆成两条,这样每次就可以走 2 k 2k 2k步了。注意到如果存在一个点 x x x,与休息点 u u u和 v v v距离都不超过 k k k,就可以从 u u u走到 v v v了。
那么可以考虑从每个休息点开始BFS出距离不超过 k k k的所有点,如果BFS过程中与其他休息点BFS到的点相交了就可以把两个休息点合并起来,用并查集维护即可。这样会得到若干个连通块,满足连通块内部任意点互相可达(进入连通块之前和出去之后甚至分别可以走不超过 k k k步不用休息),但是两个连通块内部的休息点互相不可达。
考虑一个询问 ( a , b ) (a,b) (a,b),如果 d i s ( a , b ) ≤ 2 k dis(a,b)\leq 2k dis(a,b)≤2k直接合法,否则考虑分别从 a a a向 b b b走 k k k步到达 u u u,从 b b b向 a a a走 k k k步到达 v v v,合法当且仅当 u u u和 v v v在同一个连通块内。这个充分性显然,必要性考虑你走的过程一定是先从 a a a走到一个连通块,再出来走到 b b b,其中进入之前和出去之后都不会休息,那么由于必须经过 u u u和 v v v,并且 u u u和 v v v不可能在连通块外(否则再走 k k k步也没法休息),因此 u u u和 v v v必须在同个连通块。
用倍增预处理的话时间复杂度为 O ( ( n + q ) log n ) \mathcal O((n+q)\log n) O((n+q)logn)。
#include <bits/stdc++.h>
using namespace std;
namespace SETS {
int fa[400005];
void init(int n) {
for(int i=1;i<=n;i++) fa[i]=i;
}
int find_father(int x) {
return (fa[x]==x)?x:fa[x]=find_father(fa[x]);
}
bool check(int x,int y) {
x=find_father(x);y=find_father(y);
return x!=y;
}
void