给定五个集合,问能否从五个集合各取一个元素,使得元素之和为0.
这道题有两种做法,一种是哈希,然而之前没写过哈希.....比赛后从大神那copy了一份。
这里说另一种,
对于这五个集合分为三组,1,2组求和为一组,3,4组求和分为一组,5为一组。
那么现在转化为了能否从前两组中各取一个元素,使得这两个值和为第三组一个元素的绝对值。
那么对于第一组我们升序排序,第二组我们降序排序。
对于第三组里的任一元素,假如第一组队首加第二组队首之和大于第三组的元素,那么第二组游标往后移一位,反之第一组移一位,
那么这个查找时间就为O(m),m为数组元素个数,
那么总的时间复杂度为O(n*n*n).
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<map>
#include<set>
#define eps 1e-6
#define LL long long
#define pii pair<int,int>
using namespace std;
const int maxn = 150 + 5;
//const int INF = 0x3f3f3f3f;
int n, m;
vector<int> G[maxn];
int pos[maxn];
int vis[maxn];
bool g[maxn];
int fa[maxn];
int find(int x) {
if(x == fa[x]) return x;
return fa[x] = find(fa[x]);
}
bool dfs(int u, int cur) {
if(cur == n) {
if(g[u]) {
return true;
}
else return false;
}
// cout << cur << endl;
int sz = G[u].size();
for(int i = 0; i < sz; i++) {
int v = G[u][i];
if(vis[v]) continue;
vis[v] = 1;
if(dfs(v, cur+1)) {
pos[cur+1] = v; return true;
}
vis[v] = 0;
}
return false;
}
int main() {
// freopen("input.txt", "r", stdin);
while(scanf("%d%d", &n, &m) == 2) {
for(int i = 1; i <= n; i++) G[i].clear();
memset(vis, 0, sizeof(vis));
memset(g, 0, sizeof(g));
for(int i = 1; i <= n; i++) fa[i] = 1;
while(m--) {
int u, v;
scanf("%d%d", &u, &v);
int p = find(u), q = find(v);
fa[p] = q;
G[u].push_back(v);
G[v].push_back(u);
if(u == 1) g[v] = 1;
if(v == 1) g[u] = 1;
}
int tag = 1;
for(int i = 2; i <= n; i++) {
if(find(i) != find(1)) {
tag = 0; break;
}
}
pos[1] = 1, vis[1] = 1;
if(!tag || !dfs(1, 1)) puts("no solution");
else {
for(int i = 1; i < n; i++) printf("%d ", pos[i]);
printf("%d\n", pos[n]);
}
}
return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。
原文:http://blog.csdn.net/u014664226/article/details/47189201