傳送門:codeforces 1017D
題意:給你n m q,表示在這一組數據中所有的01串長度均爲n,然後給你一個含有m個元素的multiset,之後有q次詢問。每次詢問會給你一個01串t和一個給定常數k,讓你輸出串t和multiset裏面多少個元素的“Wu”值不超過k。對於“Wu”值的定義:如果兩個01串s和t在位置i上滿足s[i]==t[i],那麼加上w[i],處理完s和t的所有n位之後的結果即爲這兩個01串的“Wu”值。
樣例解釋:第1行爲n,m,q;第2行爲w[1]...w[n],表示01串第i位的權值爲w[i];之後m行爲multiset裏的m個01串;之後q行爲q次詢問。以樣例1中的第一次詢問爲例:t=00,k=20。與multiset中的每個元素比較(假設第i個元素爲m[i]):
m[1]:01,與00相同的是第一位,所以權值s=w[1]=40;
m[2]:01,與00相同的是第一位,所以權值s=w[1]=40;
m[3]:10,與00相同的是第二位,所以權值s=w[2]=20;
m[4]:11,沒有某一位與00相同,所以權值s=0;
而上述四種比較結果中s<=20的是後兩種,所以輸出2。之後的詢問以此類推。
思路:發現m,q都特別大(1e5),而n只有1-12,即長度最長的01串只有12位,所以可以考慮狀壓來表示每一個01串的狀態,共有2^12種狀態。這樣可以通過枚舉詢問中t和multiset中s轉化爲10進制數後的值來進行預處理,而每一次要處理n位,所以複雜度爲O((2^n)*(2^n)*n),最壞情況大約是2e8。保存一個二維數組sum[4096][100],sum[i][j]表示詢問中輸入的t對應的二進制數i在與multiset中的元素s匹配時結果爲j的元素個數。詳見代碼。
AC代碼:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int w[15],cnt[4100],sum[4100][105];
void solve(){
int n,m,q;scanf("%d%d%d",&n,&m,&q);
for(int i=1;i<=n;i++) scanf("%d",&w[i]);
for(int i=1;i<=m;i++){
char s[15];scanf("%s",s);
int x=0;
for(int j=0;j<n;j++) if(s[j]=='1') x+=1<<(n-j-1);
cnt[x]++;
}
for(int i=0;i<(1<<12);i++){
for(int j=0;j<(1<<12);j++){
if(!cnt[j]) continue;
int x=0;
for(int k=0;k<12;k++){
if((i&(1<<k))==(j&(1<<k))) x+=w[n-k];
if(x>100) break;
}
if(x<=100) sum[i][x]+=cnt[j];
}
}
while(q--){
char s[15];int k;scanf("%s%d",s,&k);
int x=0;for(int i=0;i<n;i++) if(s[i]=='1') x+=1<<(n-i-1);
int ans=0;for(int i=0;i<=k;i++) ans+=sum[x][i];
printf("%d\n",ans);
}
}
int main(void){
solve();
return 0;
}