给定五个集合,问能否从五个集合各取一个元素,使得元素之和为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; }
版权声明:本文为博主原创文章,未经博主允许不得转载。
时间: 2024-10-19 07:24:06