思路是在看电视时突然想到的。枚举区间,然后按树形DP来选择最大值看是否满足条件。但枚举区间时的方法低效,看了题解,说枚举区间可以设两个指针逐步移动,
开始 l = r = 1, 记录已经出现的国家。
判断是否满足条件。
如果满足,更新答案,更新区间出现的国家,l++, 一直到不满足。
如果不满足,更新区间出现的国家,r++, 一直到满足。
但我写了好久,呃,没正确。。。。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int C,N,K,M,cnt,tot;
struct City{
int x,belong;
};
City cities[5050];
int head[2050];
int dp[2050][2];
bool vis[2050];
struct node{
int u,v;
int next;
}edge[6500];
bool cmp(City a,City b){
if(a.x<b.x) return true;
return false;
}
int belong[2050];
void addedge(int u,int v){
edge[tot].u=u;
edge[tot].v=v;
edge[tot].next=head[u];
head[u]=tot++;
}
void dfs(int u,int f){
dp[u][0]=0;dp[u][1]=1;
vis[u]=true;
for(int ei=head[u];ei!=-1;ei=edge[ei].next){
if(f!=edge[ei].v&&belong[edge[ei].v]>0&&!vis[edge[ei].v ]){
dfs(edge[ei].v,u);
dp[u][1]+=dp[edge[ei].v][0];
dp[u][0]+=max(dp[edge[ei].v][0],dp[edge[ei].v][1]);
}
}
}
bool judge(){
int ans=0;
memset(vis,false,sizeof(vis));
for(int i=1;i<=N;i++){
if(!vis[i]&&belong[i]>0){
dfs(i,0);
ans+=max(dp[i][0],dp[i][1]);
}
}
if(ans>=K) return true;
else return false;
}
int main(){
int T,t=0,u,v,l,r;
scanf("%d",&T);
while(++t<=T){
scanf("%d%d%d%d",&C,&N,&K,&M);
cnt=tot=0;
for(int i=1;i<=C;i++){
scanf("%d%d",&cities[i].x,&cities[i].belong);
}
memset(head,-1,sizeof(head));
sort(cities+1,cities+1+C,cmp);
for(int i=1;i<=M;i++){
scanf("%d%d",&u,&v);
addedge(u,v);
addedge(v,u);
}
cnt=l=r=1;
int ans=-1;
memset(belong,0,sizeof(belong));
while(r<=C){
if(cnt>=K && judge()){
if(ans==-1||cities[r].x - cities[l].x<ans)
ans=cities[r].x - cities[l].x;
belong[ cities[l].belong ]--;
if( belong[ cities[l].belong ]==0 ) cnt--;
l++;
}
else{
r++;
belong[ cities[r].belong ]++;
if( belong[ cities[r].belong ]==1 ) cnt++;
}
}
printf("Case #%d: %d\n",t,ans);
}
return 0;
}
下面是别人过了的代码。。。T_T不知道错哪里了
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int C = 5005;
const int N = 2005;
const int M = 1005;
int cnt[N]; //i????????
int vis[C];
int dp[C][2];
int c, n, k, m;
vector<int>v[N];
struct city
{
int x, y;
bool operator<(const city &a) const
{
return x<a.x;
}
}ct[C];
int min(int x, int y)
{
if(x==-1) return y;
return x<y?x:y;
}
void dfs(int s)
{
vis[s] = 1;
dp[s][1] = 1;
dp[s][0] = 0;
for(int i=0; i<v[s].size(); i++)
{
int ss = v[s][i];
if(vis[ss]==1 || cnt[ss]==0) continue;
dfs(ss);
dp[s][1] += dp[ss][0];
dp[s][0] += max(dp[ss][1], dp[ss][0]);
}
}
bool ok(int l, int r)
{
int res = 0;
memset(vis, 0, sizeof(vis));
for(int i = l; i<=r; i++)
if(vis[ ct[i].y ]==0)
{
dfs( ct[i].y );
res += max( dp[ ct[i].y ][0], dp[ ct[i].y ][1] );
}
if(res>=k) return true;
else return false;
}
int solve()
{
int kk, l, r;
kk = l = r = 1;
cnt[ ct[1].y ]++;
int ans = -1;
while(r<=c)
{
if(kk>=k && ok(l, r))
{
ans = min(ans, ct[r].x - ct[l].x);
cnt[ ct[l].y ]--;
if( cnt[ ct[l].y ]==0 ) kk--;
l++;
}
else
{
r++;
cnt[ ct[r].y ]++;
if( cnt[ ct[r].y ]==1 ) kk++;
}
}
return ans;
}
int main()
{
int t, tt=0, i, x, y;
scanf("%d", &t);
while(t--)
{
scanf("%d%d%d%d", &c, &n, &k, &m);
for(i=1; i<=n; i++) v[i].clear();
memset(cnt, 0, sizeof(cnt));
memset(vis, 0, sizeof(vis));
for(i=1; i<=c; i++) scanf("%d%d", &ct[i].x, &ct[i].y);
sort(ct+1, ct+1+c);
for(i=1; i<=m; i++)
{
scanf("%d%d", &x, &y);
v[x].push_back(y);
v[y].push_back(x);
}
printf("Case #%d: %d\n", ++tt, solve());
}
return 0;
}