L2-025 分而治之 - 思维

分而治之,各个击破是兵家常用的策略之一。在战争中,我们希望首先攻下敌方的部分城市,使其剩余的城市变成孤立无援,然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序,判断每个方案的可行性。

输入格式:

输入在第一行给出两个正整数 N 和 M(均不超过10 000),分别为敌方城市个数(于是默认城市从 1 到 N 编号)和连接两城市的通路条数。随后 M 行,每行给出一条通路所连接的两个城市的编号,其间以一个空格分隔。在城市信息之后给出参谋部的系列方案,即一个正整数 K (≤ 100)和随后的 K 行方案,每行按以下格式给出:

Np v[1] v[2] ... v[Np]

其中 Np 是该方案中计划攻下的城市数量,后面的系列 v[i] 是计划攻下的城市编号。

输出格式:

对每一套方案,如果可行就输出YES,否则输出NO

输入样例:

10 11
8 7
6 8
4 5
8 4
8 1
1 2
1 4
9 8
9 1
1 10
2 4
5
4 10 3 8 4
6 6 1 7 5 4 9
3 1 8 4
2 2 8
7 9 8 7 6 5 4 2

输出样例:

NO
YES
YES
NO
NO

思路:

刚开始想的是暴力,记录每个城市连得边数,然后两个for循环看看摧毁的城市里有没有连着的边,最后摧毁的总边,判断是否小于总边数

但是超时了,= =

后来搜了搜

思路还是两个for循环,用数组编辑城市是否摧毁,若城市没摧毁,并且它相连的城市有没有摧毁的,就不成功。

代码如下:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef pair<int,int>P;
const int INF=0x3f3f3f3f;
const int N=10015,mod=32767;
vector<int>co[N];
int n,d[N];

int solve(){
    for(int i=1;i<=n;i++){
        if(d[i])continue;
        for(int j=0;j<co[i].size();j++){
            int ci=co[i][j];
            if(!d[ci])return 0;
        }
    }
    return 1;
}

int main(){
    int m,a,b;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&a,&b);
        co[a].push_back(b);
        co[b].push_back(a);
    }
    int k,p,t;
    scanf("%d",&k);
    while(k--){
        scanf("%d",&p);
        memset(d,0,sizeof(d));
        for(int i=1;i<=p;i++){
            scanf("%d",&t);
            d[t]=1;
        }
        int flag=solve();
        if(!flag)printf("NO\n");
        else printf("YES\n");
    }
}

超时代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef pair<int,int>P;
const int INF=0x3f3f3f3f;
const int N=10015,mod=32767;
int tmp[N],tot[N];
set<int>co[N];


int main(){
    int n,m,a,b;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&a,&b);
        tot[a]++;tot[b]++;
        co[a].insert(b);
        co[b].insert(a);
    }
    int k,p,cnt;
    scanf("%d",&k);
    while(k--){
        scanf("%d",&p);
        cnt=0;
        for(int i=1;i<=p;i++){
            scanf("%d",&tmp[i]);
            cnt+=tot[tmp[i]];
        }
        if(cnt<m){printf("NO\n");continue;}
        int flag=0;
        for(int i=1;i<p;i++){
            for(int j=i+1;j<=p;j++){
                int a=tmp[i],b=tmp[j];
                if(co[a].find(b)!=co[a].end())cnt--;
                if(cnt<m){flag=1;break;}
            }
            if(flag)break;
        }
        if(flag)printf("NO\n");
        else printf("YES\n");
    }
}

(可能是因为,正确的方法标记了摧毁的城市(剪枝),而超时代码没能剪枝,还需要在set中查找……多动动脑子,想想到底什么情况下还有相连的城市)

猜你喜欢

转载自blog.csdn.net/m0_37579232/article/details/87914657