之前有做过题目,用vector存取超时了,但是没记录,忘记是哪道了…不过今天遇到了一道哈哈哈,记录一下。
题目:https://codeforces.com/gym/102396/problem/G
题意:有n个砝码,重各为a[i]。现有天平,称重是将其两个盘子上的重量各自%m再比较。问如何安放砝码,这特殊的天平能平衡。
具体思路很简单,看代码就知道了。不过要记得如果两种分配同时拥有同一个砝码时,这个砝码直接踢出去,就不会影响了,毕竟砝码都正。(因为这个半天不敢下手,敢下手的时候比赛都快结束了)
vector会超时:
#include <cstdio>
#include <cstring>
#include<iostream>
#include <algorithm>
#include <map>
#include<vector>
#include<stack>
using namespace std;
typedef long long ll;
const int maxn=(1<<26)+6;
int a[40];
vector<int>g[maxn];
int n,m;
void print(int n1,int n2){
vector<int>A;
vector<int>B;
for(int i=0;i<n;i++){
if((n1&(1<<i))&&!(n2&(1<<i)))A.push_back(i+1);
if((n2&(1<<i))&&!(n1&(1<<i)))B.push_back(i+1);
}
printf("%d\n",A.size());
for(int i=0;i<A.size();i++)printf("%d ",A[i]);printf("\n");
printf("%d\n",B.size());
for(int i=0;i<B.size();i++)printf("%d ",B[i]);printf("\n");
}
int t;
void dfs(int x,int sum,int rec){
if(x>n)return;
for(int i=0;i<g[sum].size();i++){
int u=g[sum][i];
if(u==rec)continue;
print(u,rec);
exit(0);
}
g[sum].push_back(rec);
dfs(x+1,sum,rec);
dfs(x+1,(sum+a[x])%m,rec+(1<<x));
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
dfs(0,0,0);
printf("-1\n");
}
数组则不会:
#include <cstdio>
#include <cstring>
#include<iostream>
#include <algorithm>
#include <map>
#include<vector>
#include<stack>
using namespace std;
typedef long long ll;
const int maxn=(1<<26)+6;
int a[40];
int g[maxn];
int n,m;
void print(int n1,int n2){
vector<int>A;
vector<int>B;
for(int i=0;i<n;i++){
if((n1&(1<<i))&&!(n2&(1<<i)))A.push_back(i+1);
if((n2&(1<<i))&&!(n1&(1<<i)))B.push_back(i+1);
}
printf("%d\n",A.size());
for(int i=0;i<A.size();i++)printf("%d ",A[i]);printf("\n");
printf("%d\n",B.size());
for(int i=0;i<B.size();i++)printf("%d ",B[i]);printf("\n");
}
int t;
void dfs(int x,int sum,int rec){
if(x>n)return;
if(g[sum]!=-1&&g[sum]!=rec){
print(g[sum],rec);
exit(0);
}
g[sum]=rec;
dfs(x+1,sum,rec);
dfs(x+1,(sum+a[x])%m,rec+(1<<x));
}
int main()
{
scanf("%d%d",&n,&m);
memset(g,-1,sizeof(g));
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
dfs(0,0,0);
printf("-1\n");
}
这两个代码不同之处仅在于g的定义。隐约记得看过资料说vector的存取貌似有一定的时间(?)。
平时很喜欢用vector就是我本人了,把坑记下!!!